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],
|