arjunanand13 commited on
Commit
43c92dd
1 Parent(s): 921d4a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ class VideoClassifier:
42
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
43
  if self.model_name=='mistral':
44
  print("Setting up Mistral model for Class Selection")
45
- self.setup_mistral_model()
46
  else :
47
  print("Setting up Gemini model for Class Selection")
48
  self.setup_gemini_model()
 
42
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
43
  if self.model_name=='mistral':
44
  print("Setting up Mistral model for Class Selection")
45
+ self.setup_mistral_space_model()
46
  else :
47
  print("Setting up Gemini model for Class Selection")
48
  self.setup_gemini_model()