prithivMLmods commited on
Commit
20057cc
·
verified ·
1 Parent(s): 50bbc28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -193,14 +193,14 @@ def process_document_stream(
193
  yield "Please enter a prompt.", ""
194
  return
195
 
196
- if model_name == "LFM2-VL-450M": processor, model = processor_m, model_m
197
- elif model_name == "LFM2-VL-1.6B": processor, model = processor_t, model_t
198
- elif model_name == "ShotVL-3B": processor, model = processor_z, model_z
199
- elif model_name == "SmolVLM-Instruct-250M": processor, model = processor_c, model_c
200
- elif model_name == "MonkeyOCR-pro-1.2B": processor, model = processor_g, model_g
201
- elif model_name == "VLAA-Thinker-Qwen2VL-2B": processor, model = processor_i, model_i
202
- elif model_name == "Nanonets-OCR-s": processor, model = processor_a, model_a
203
- elif model_name == "Megalodon-OCR-Sync-0713": processor, model = processor_x, model_x
204
  else:
205
  yield "Invalid model selected.", ""
206
  return
@@ -257,8 +257,8 @@ def create_gradio_interface():
257
  # Left Column (Inputs)
258
  with gr.Column(scale=1):
259
  model_choice = gr.Dropdown(
260
- choices=["LFM2-VL-1.6B", "LFM2-VL-450M", "SmolVLM-Instruct-250M", "ShotVL-3B", "Megalodon-OCR-Sync-0713", "VLAA-Thinker-Qwen2VL-2B", "MonkeyOCR-pro-1.2B",
261
- "Nanonets-OCR-s"],
262
  label="Select Model", value="LFM2-VL-1.6B"
263
  )
264
  prompt_input = gr.Textbox(label="Query Input", placeholder="✦︎ Enter your query", value="Describe the image!")
 
193
  yield "Please enter a prompt.", ""
194
  return
195
 
196
+ if model_name == "LFM2-VL-450M(fast)": processor, model = processor_m, model_m
197
+ elif model_name == "LFM2-VL-1.6B(fast)": processor, model = processor_t, model_t
198
+ elif model_name == "ShotVL-3B(cinematic)": processor, model = processor_z, model_z
199
+ elif model_name == "SmolVLM-Instruct-250M(smol)": processor, model = processor_c, model_c
200
+ elif model_name == "MonkeyOCR-pro-1.2B(ocr)": processor, model = processor_g, model_g
201
+ elif model_name == "VLAA-Thinker-Qwen2VL-2B(reason)": processor, model = processor_i, model_i
202
+ elif model_name == "Nanonets-OCR-s(ocr)": processor, model = processor_a, model_a
203
+ elif model_name == "Megalodon-OCR-Sync-0713(ocr)": processor, model = processor_x, model_x
204
  else:
205
  yield "Invalid model selected.", ""
206
  return
 
257
  # Left Column (Inputs)
258
  with gr.Column(scale=1):
259
  model_choice = gr.Dropdown(
260
+ choices=["LFM2-VL-1.6B(fast)", "LFM2-VL-450M(fast)", "SmolVLM-Instruct-250M(smol)", "ShotVL-3B(cinematic)", "Megalodon-OCR-Sync-0713(ocr)", "VLAA-Thinker-Qwen2VL-2B(reason)", "MonkeyOCR-pro-1.2B(ocr)",
261
+ "Nanonets-OCR-s(ocr)"],
262
  label="Select Model", value="LFM2-VL-1.6B"
263
  )
264
  prompt_input = gr.Textbox(label="Query Input", placeholder="✦︎ Enter your query", value="Describe the image!")