arjunanand13 commited on
Commit
2b03cfe
·
verified ·
1 Parent(s): 31f6d21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -423,10 +423,11 @@ class VideoClassifier:
423
  return final_answer
424
 
425
 
426
- def save_model_choice(self,model_name="mistral"):
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")