Spaces:
Runtime error
Runtime error
Commit
•
a0af599
1
Parent(s):
cca1ded
Update app.py
Browse files
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
|