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

Update app.py

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