arjunanand13 commited on
Commit
318e88f
1 Parent(s): 397014d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,7 +63,8 @@ class VideoClassifier:
63
  # if not self.hf_key:
64
  # raise ValueError("Hugging Face API key is not set or invalid.")
65
 
66
- self.client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
 
67
  # self.client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
68
  self.whisper_model = whisper.load_model("base")
69
  self.img_cap = Caption()
 
63
  # if not self.hf_key:
64
  # raise ValueError("Hugging Face API key is not set or invalid.")
65
 
66
+ # self.client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
67
+ self.client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
68
  # self.client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
69
  self.whisper_model = whisper.load_model("base")
70
  self.img_cap = Caption()