prithivMLmods commited on
Commit
a126e19
·
verified ·
1 Parent(s): ff89480

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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("echo840/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,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: Changed pro.Chatbot.WelcomeConfig to a dictionary
348
  welcome_config={
349
- "prompts": welcome_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
  )