Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -231,12 +231,12 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 231 |
repetition_penalty = gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)
|
| 232 |
with gr.Column():
|
| 233 |
output = gr.Textbox(label="Output", interactive=False)
|
| 234 |
-
|
| 235 |
-
|
| 236 |
-
|
| 237 |
-
|
| 238 |
-
|
| 239 |
-
|
| 240 |
image_submit.click(
|
| 241 |
fn=generate_image,
|
| 242 |
inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
|
|
|
|
| 231 |
repetition_penalty = gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)
|
| 232 |
with gr.Column():
|
| 233 |
output = gr.Textbox(label="Output", interactive=False)
|
| 234 |
+
with gr.row():
|
| 235 |
+
model_choice = gr.Dropdown(
|
| 236 |
+
choices=["Cosmos-Reason1-7B", "MiMo-VL-7B-RL"],
|
| 237 |
+
label="Select Model",
|
| 238 |
+
value="Cosmos-Reason1-7B")
|
| 239 |
+
|
| 240 |
image_submit.click(
|
| 241 |
fn=generate_image,
|
| 242 |
inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
|