Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ def load_model(model_id, processor_class, model_class, subfolder=None, model_key
|
|
| 64 |
|
| 65 |
# Load all models
|
| 66 |
load_model("nanonets/Nanonets-OCR-s", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Nanonets-OCR-s")
|
| 67 |
-
load_model("
|
| 68 |
load_model("scb10x/typhoon-ocr-7b", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Typhoon-OCR-7B")
|
| 69 |
load_model("ds4sd/SmolDocling-256M-preview", AutoProcessor, AutoModelForVision2Seq, model_key="SmolDocling-256M-preview")
|
| 70 |
load_model("Kwai-Keye/Thyme-RL", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Thyme-RL")
|
|
@@ -344,9 +344,11 @@ with gr.Blocks(css=css, fill_width=True, title="Multimodal OCR2") as demo:
|
|
| 344 |
gr.Markdown("## Multimodal OCR2")
|
| 345 |
chatbot = pro.Chatbot(
|
| 346 |
height="calc(100vh - 200px)",
|
| 347 |
-
# FIX:
|
| 348 |
welcome_config={
|
| 349 |
-
"prompts":
|
|
|
|
|
|
|
| 350 |
"title": "Start by selecting an example:"
|
| 351 |
}
|
| 352 |
)
|
|
|
|
| 64 |
|
| 65 |
# Load all models
|
| 66 |
load_model("nanonets/Nanonets-OCR-s", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Nanonets-OCR-s")
|
| 67 |
+
load_model("echo80/MonkeyOCR", AutoProcessor, Qwen2_5_VLForConditionalGeneration, subfolder="Recognition", model_key="MonkeyOCR-Recognition")
|
| 68 |
load_model("scb10x/typhoon-ocr-7b", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Typhoon-OCR-7B")
|
| 69 |
load_model("ds4sd/SmolDocling-256M-preview", AutoProcessor, AutoModelForVision2Seq, model_key="SmolDocling-256M-preview")
|
| 70 |
load_model("Kwai-Keye/Thyme-RL", AutoProcessor, Qwen2_5_VLForConditionalGeneration, model_key="Thyme-RL")
|
|
|
|
| 344 |
gr.Markdown("## Multimodal OCR2")
|
| 345 |
chatbot = pro.Chatbot(
|
| 346 |
height="calc(100vh - 200px)",
|
| 347 |
+
# FIX: The `prompts` key now holds a dictionary for categorization
|
| 348 |
welcome_config={
|
| 349 |
+
"prompts": {
|
| 350 |
+
"Examples": welcome_prompts
|
| 351 |
+
},
|
| 352 |
"title": "Start by selecting an example:"
|
| 353 |
}
|
| 354 |
)
|