prithivMLmods commited on
Commit
cb3bf4e
·
verified ·
1 Parent(s): 2144b6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
79
  if model_name == "Cosmos-Reason1-7B":
80
  processor = processor_m
81
  model = model_m
82
- elif model_name == "MiMo-VL-7B-RL":
83
  processor = processor_x
84
  model = model_x
85
  else:
 
79
  if model_name == "Cosmos-Reason1-7B":
80
  processor = processor_m
81
  model = model_m
82
+ elif model_name == "MiMo-VL-7B-SFT":
83
  processor = processor_x
84
  model = model_x
85
  else: