Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -305,7 +305,7 @@ class VideoClassifier:
|
|
| 305 |
# return final_answer, first_video, second_video
|
| 306 |
return final_answer
|
| 307 |
|
| 308 |
-
def save_model_choice(self,model_name):
|
| 309 |
self.model_name = model_name
|
| 310 |
if self.model_name=='mistral':
|
| 311 |
print("Setting up Mistral model for Class Selection")
|
|
|
|
| 305 |
# return final_answer, first_video, second_video
|
| 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")
|