Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -101,7 +101,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 101 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 102 |
processor = processor_x
|
| 103 |
model = model_x
|
| 104 |
-
elif model_name == "Captioner-7B":
|
| 105 |
processor = processor_z
|
| 106 |
model = model_z
|
| 107 |
elif model_name == "visionOCR-3B":
|
|
@@ -158,7 +158,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
| 158 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 159 |
processor = processor_x
|
| 160 |
model = model_x
|
| 161 |
-
elif model_name == "Captioner-7B":
|
| 162 |
processor = processor_z
|
| 163 |
model = model_z
|
| 164 |
elif model_name == "visionOCR-3B":
|
|
@@ -274,7 +274,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 274 |
markdown_output = gr.Markdown()
|
| 275 |
|
| 276 |
model_choice = gr.Radio(
|
| 277 |
-
choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B", "visionOCR-3B"],
|
| 278 |
label="Select Model",
|
| 279 |
value="Cosmos-Reason1-7B"
|
| 280 |
)
|
|
|
|
| 101 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 102 |
processor = processor_x
|
| 103 |
model = model_x
|
| 104 |
+
elif model_name == "Captioner-7B-Qwen2.5VL":
|
| 105 |
processor = processor_z
|
| 106 |
model = model_z
|
| 107 |
elif model_name == "visionOCR-3B":
|
|
|
|
| 158 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 159 |
processor = processor_x
|
| 160 |
model = model_x
|
| 161 |
+
elif model_name == "Captioner-7B-Qwen2.5VL":
|
| 162 |
processor = processor_z
|
| 163 |
model = model_z
|
| 164 |
elif model_name == "visionOCR-3B":
|
|
|
|
| 274 |
markdown_output = gr.Markdown()
|
| 275 |
|
| 276 |
model_choice = gr.Radio(
|
| 277 |
+
choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B-Qwen2.5VL", "visionOCR-3B"],
|
| 278 |
label="Select Model",
|
| 279 |
value="Cosmos-Reason1-7B"
|
| 280 |
)
|