arjunanand13 commited on
Commit
7997dd9
1 Parent(s): 03396aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ class VideoClassifier:
34
  caption_time = 0
35
  classification_time = 0
36
  def __init__(self, no_of_frames, mode='interface',model='mistral'):
37
-
38
  self.no_of_frames = no_of_frames
39
  self.mode = mode
40
  self.model_name = model.strip().lower()
@@ -47,7 +47,7 @@ class VideoClassifier:
47
  print("Setting up Gemini model for Class Selection")
48
  self.setup_gemini_model()
49
  self.setup_paths()
50
- self.hf_key = os.environ.get("HF_KEY", None)
51
  """chatgpt 3.5"""
52
  # self.chatgpt_client = OpenAI(api_key="sk-proj-KY1qI7zTpsUiJhMUHuNdT3BlbkFJLOjVnTUSpYJi87yUtSEI")
53
  self.chatgpt_client= OpenAI(api_key="sk-proj-TVoFQ4X9apDUs0V6zCDIT3BlbkFJmWRNMgJ6fapge12zygzG")
 
34
  caption_time = 0
35
  classification_time = 0
36
  def __init__(self, no_of_frames, mode='interface',model='mistral'):
37
+ self.hf_key = os.environ.get("HF_KEY", None)
38
  self.no_of_frames = no_of_frames
39
  self.mode = mode
40
  self.model_name = model.strip().lower()
 
47
  print("Setting up Gemini model for Class Selection")
48
  self.setup_gemini_model()
49
  self.setup_paths()
50
+ # self.hf_key = os.environ.get("HF_KEY", None)
51
  """chatgpt 3.5"""
52
  # self.chatgpt_client = OpenAI(api_key="sk-proj-KY1qI7zTpsUiJhMUHuNdT3BlbkFJLOjVnTUSpYJi87yUtSEI")
53
  self.chatgpt_client= OpenAI(api_key="sk-proj-TVoFQ4X9apDUs0V6zCDIT3BlbkFJmWRNMgJ6fapge12zygzG")