Update app.py

app.py

@@ -77,19 +77,19 @@ class VideoClassifier:
         print(hf_auth)
         model_config = transformers.AutoConfig.from_pretrained(
             self.model_id,
-            use_auth_token=hf_auth
+            # use_auth_token=hf_auth
         )
         self.model = transformers.AutoModelForCausalLM.from_pretrained(
             self.model_id,
             trust_remote_code=True,
             config=model_config,
             quantization_config=bnb_config,
-            use_auth_token=hf_auth
+            # use_auth_token=hf_auth
         )
         self.model.eval()
         self.tokenizer = transformers.AutoTokenizer.from_pretrained(
             self.model_id,
-            use_auth_token=hf_auth
+            # use_auth_token=hf_auth
         )
         self.generate_text = transformers.pipeline(
             model=self.model, tokenizer=self.tokenizer,
@@ -308,7 +308,7 @@ class VideoClassifier:
         self.model_name = model_name
         if self.model_name=='mistral':
             print("Setting up Mistral model for Class Selection")
-            self.setup_mistral_space_model()
+            self.setup_mistral_model()
         else :
             print("Setting up Gemini model for Class Selection")
             self.setup_gemini_model()