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, lines=2, scale=2)
|
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 == "Captioner-7B":
|
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 == "Captioner-7B":
|
149 |
processor = processor_z
|
150 |
model = model_z
|
151 |
else:
|
|
|
246 |
with gr.Column():
|
247 |
output = gr.Textbox(label="Output", interactive=False, lines=2, scale=2)
|
248 |
model_choice = gr.Radio(
|
249 |
+
choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B"],
|
250 |
label="Select Model",
|
251 |
value="Cosmos-Reason1-7B"
|
252 |
)
|