Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -91,7 +91,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 91 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 92 |
processor = processor_x
|
| 93 |
model = model_x
|
| 94 |
-
elif model_name == "
|
| 95 |
processor = processor_z
|
| 96 |
model = model_z
|
| 97 |
else:
|
|
@@ -145,7 +145,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
| 145 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 146 |
processor = processor_x
|
| 147 |
model = model_x
|
| 148 |
-
elif model_name == "
|
| 149 |
processor = processor_z
|
| 150 |
model = model_z
|
| 151 |
else:
|
|
@@ -246,7 +246,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 246 |
with gr.Column():
|
| 247 |
output = gr.Textbox(label="Output", interactive=False)
|
| 248 |
model_choice = gr.Radio(
|
| 249 |
-
choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "
|
| 250 |
label="Select Model",
|
| 251 |
value="Cosmos-Reason1-7B"
|
| 252 |
)
|
|
|
|
| 91 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 92 |
processor = processor_x
|
| 93 |
model = model_x
|
| 94 |
+
elif model_name == "Relaxed7B":
|
| 95 |
processor = processor_z
|
| 96 |
model = model_z
|
| 97 |
else:
|
|
|
|
| 145 |
elif model_name == "docscopeOCR-7B-050425-exp":
|
| 146 |
processor = processor_x
|
| 147 |
model = model_x
|
| 148 |
+
elif model_name == "Relaxed7B":
|
| 149 |
processor = processor_z
|
| 150 |
model = model_z
|
| 151 |
else:
|
|
|
|
| 246 |
with gr.Column():
|
| 247 |
output = gr.Textbox(label="Output", interactive=False)
|
| 248 |
model_choice = gr.Radio(
|
| 249 |
+
choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Relaxed7B"],
|
| 250 |
label="Select Model",
|
| 251 |
value="Cosmos-Reason1-7B"
|
| 252 |
)
|