Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ MODEL_PATHS = {
|
|
| 28 |
"Model 1 (Complex handwrittings )": ("prithivMLmods/Qwen2.5-VL-7B-Abliterated-Caption-it", Qwen2_5_VLForConditionalGeneration),
|
| 29 |
"Model 2 (simple and scanned handwritting )": ("nanonets/Nanonets-OCR-s", Qwen2_5_VLForConditionalGeneration),
|
| 30 |
}
|
| 31 |
-
|
| 32 |
|
| 33 |
MAX_NEW_TOKENS_DEFAULT = 512
|
| 34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
@@ -460,7 +460,7 @@ def _write_trainer_script():
|
|
| 460 |
|
| 461 |
# ---------------- Gradio Interface ----------------
|
| 462 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 463 |
-
gr.Markdown("## ✍🏾 wilson Handwritten
|
| 464 |
|
| 465 |
model_choice = gr.Radio(choices=list(MODEL_PATHS.keys()),
|
| 466 |
value=list(MODEL_PATHS.keys())[0],
|
|
|
|
| 28 |
"Model 1 (Complex handwrittings )": ("prithivMLmods/Qwen2.5-VL-7B-Abliterated-Caption-it", Qwen2_5_VLForConditionalGeneration),
|
| 29 |
"Model 2 (simple and scanned handwritting )": ("nanonets/Nanonets-OCR-s", Qwen2_5_VLForConditionalGeneration),
|
| 30 |
}
|
| 31 |
+
|
| 32 |
|
| 33 |
MAX_NEW_TOKENS_DEFAULT = 512
|
| 34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 460 |
|
| 461 |
# ---------------- Gradio Interface ----------------
|
| 462 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 463 |
+
gr.Markdown("## ✍🏾 wilson Handwritten text recognition with Feedback Loop")
|
| 464 |
|
| 465 |
model_choice = gr.Radio(choices=list(MODEL_PATHS.keys()),
|
| 466 |
value=list(MODEL_PATHS.keys())[0],
|