arjunanand13 commited on
Commit
a0af599
1 Parent(s): cca1ded

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,6 +36,7 @@ class VideoClassifier:
36
  self.setup_gemini_model()
37
  self.setup_paths()
38
  self.hf_key = os.environ.get("HF_KEY", None)
 
39
  def setup_paths(self):
40
  self.path = './results'
41
  if os.path.exists(self.path):
@@ -60,7 +61,7 @@ class VideoClassifier:
60
  bnb_4bit_use_double_quant=True,
61
  bnb_4bit_compute_dtype=bfloat16,
62
  )
63
- hf_auth = hf_key
64
  model_config = transformers.AutoConfig.from_pretrained(
65
  self.model_id,
66
  use_auth_token=hf_auth
 
36
  self.setup_gemini_model()
37
  self.setup_paths()
38
  self.hf_key = os.environ.get("HF_KEY", None)
39
+
40
  def setup_paths(self):
41
  self.path = './results'
42
  if os.path.exists(self.path):
 
61
  bnb_4bit_use_double_quant=True,
62
  bnb_4bit_compute_dtype=bfloat16,
63
  )
64
+ hf_auth = self.hf_key
65
  model_config = transformers.AutoConfig.from_pretrained(
66
  self.model_id,
67
  use_auth_token=hf_auth