Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -306,7 +306,7 @@ class VideoClassifier:
|
|
306 |
return final_answer
|
307 |
|
308 |
def save_model_choice(self,model_name='gemini'):
|
309 |
-
self.model_name = model_name
|
310 |
if self.model_name=='mistral':
|
311 |
print("Setting up Mistral model for Class Selection")
|
312 |
self.setup_mistral_space_model()
|
|
|
306 |
return final_answer
|
307 |
|
308 |
def save_model_choice(self,model_name='gemini'):
|
309 |
+
self.model_name = model_name or 'gemini'
|
310 |
if self.model_name=='mistral':
|
311 |
print("Setting up Mistral model for Class Selection")
|
312 |
self.setup_mistral_space_model()
|