Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -423,10 +423,11 @@ class VideoClassifier:
|
|
| 423 |
return final_answer
|
| 424 |
|
| 425 |
|
| 426 |
-
def save_model_choice(self,model_name
|
| 427 |
global setup_time
|
| 428 |
start_time_setup = time.time()
|
| 429 |
-
|
|
|
|
| 430 |
self.model_name = model_name
|
| 431 |
if self.model_name=='mistral':
|
| 432 |
print("Setting up Mistral model for Class Selection")
|
|
|
|
| 423 |
return final_answer
|
| 424 |
|
| 425 |
|
| 426 |
+
def save_model_choice(self,model_name):
|
| 427 |
global setup_time
|
| 428 |
start_time_setup = time.time()
|
| 429 |
+
if not model_name:
|
| 430 |
+
model_name = "mistral"
|
| 431 |
self.model_name = model_name
|
| 432 |
if self.model_name=='mistral':
|
| 433 |
print("Setting up Mistral model for Class Selection")
|