prithivMLmods commited on
Commit
40b4f90
·
verified ·
1 Parent(s): 19a90b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 == "Relaxed-7B":
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 == "Relaxed-7B":
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", "Relaxed-7B"],
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
  )