prithivMLmods commited on
Commit
56e2149
·
verified ·
1 Parent(s): 25efeb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -209,10 +209,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
209
  with gr.TabItem("Image Inference"):
210
  image_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
211
  image_upload = gr.Image(type="pil", label="Image")
212
- model_choice = gr.Dropdown(
213
- choices=["Cosmos-Reason1-7B", "MiMo-VL-7B-RL"],
214
- label="Select Model",
215
- value="Cosmos-Reason1-7B")
216
  image_submit = gr.Button("Submit", elem_classes="submit-btn")
217
  gr.Examples(
218
  examples=image_examples,
@@ -221,10 +217,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
221
  with gr.TabItem("Video Inference"):
222
  video_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
223
  video_upload = gr.Video(label="Video")
224
- model_choice = gr.Dropdown(
225
- choices=["Cosmos-Reason1-7B", "MiMo-VL-7B-RL"],
226
- label="Select Model",
227
- value="Cosmos-Reason1-7B")
228
  video_submit = gr.Button("Submit", elem_classes="submit-btn")
229
  gr.Examples(
230
  examples=video_examples,
@@ -239,7 +231,13 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
239
  repetition_penalty = gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)
240
  with gr.Column():
241
  output = gr.Textbox(label="Output", interactive=False)
242
-
 
 
 
 
 
 
243
  image_submit.click(
244
  fn=generate_image,
245
  inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
 
209
  with gr.TabItem("Image Inference"):
210
  image_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
211
  image_upload = gr.Image(type="pil", label="Image")
 
 
 
 
212
  image_submit = gr.Button("Submit", elem_classes="submit-btn")
213
  gr.Examples(
214
  examples=image_examples,
 
217
  with gr.TabItem("Video Inference"):
218
  video_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
219
  video_upload = gr.Video(label="Video")
 
 
 
 
220
  video_submit = gr.Button("Submit", elem_classes="submit-btn")
221
  gr.Examples(
222
  examples=video_examples,
 
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
+ model_choice = gr.Dropdown(
236
+ choices=["Cosmos-Reason1-7B", "MiMo-VL-7B-RL"],
237
+ label="Select Model",
238
+ value="Cosmos-Reason1-7B"
239
+ )
240
+
241
  image_submit.click(
242
  fn=generate_image,
243
  inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],