Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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!")
|