arjunanand13 commited on
Commit
145b57b
·
verified ·
1 Parent(s): 4a351ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")