Spaces:
Running
on
A100
Running
on
A100
adding auth
Browse files
app.py
CHANGED
@@ -10,7 +10,37 @@ from pathlib import Path
|
|
10 |
import time
|
11 |
import torch
|
12 |
import spaces
|
13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
|
15 |
from video_highlight_detector import (
|
16 |
load_model,
|
@@ -179,5 +209,9 @@ if __name__ == "__main__":
|
|
179 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
180 |
zero = torch.Tensor([0]).to(device)
|
181 |
|
|
|
|
|
|
|
|
|
182 |
app = create_ui("video_spec.json")
|
183 |
app.launch()
|
|
|
10 |
import time
|
11 |
import torch
|
12 |
import spaces
|
13 |
+
import os
|
14 |
+
from huggingface_hub import HfApi, login
|
15 |
+
from dotenv import load_dotenv
|
16 |
+
|
17 |
+
def setup_hf_auth():
|
18 |
+
"""
|
19 |
+
Set up Hugging Face authentication using a token stored as a secret.
|
20 |
+
Returns authenticated HfApi instance if successful.
|
21 |
+
"""
|
22 |
+
try:
|
23 |
+
# First try to get token from environment variable
|
24 |
+
hf_token = os.getenv('HF_TOKEN')
|
25 |
+
|
26 |
+
if not hf_token:
|
27 |
+
raise ValueError("HF_TOKEN not found in environment variables or .env file")
|
28 |
+
|
29 |
+
# Login to Hugging Face
|
30 |
+
login(token=hf_token)
|
31 |
+
|
32 |
+
# Initialize API client
|
33 |
+
api = HfApi()
|
34 |
+
|
35 |
+
# Verify authentication by trying to get user info
|
36 |
+
user_info = api.whoami()
|
37 |
+
print(f"Successfully authenticated as: {user_info.name}")
|
38 |
+
|
39 |
+
return api
|
40 |
+
|
41 |
+
except Exception as e:
|
42 |
+
print(f"Authentication failed: {str(e)}")
|
43 |
+
return None
|
44 |
|
45 |
from video_highlight_detector import (
|
46 |
load_model,
|
|
|
209 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
210 |
zero = torch.Tensor([0]).to(device)
|
211 |
|
212 |
+
hf_api = setup_hf_auth()
|
213 |
+
if not hf_api:
|
214 |
+
raise RuntimeError("Failed to authenticate with Hugging Face")
|
215 |
+
|
216 |
app = create_ui("video_spec.json")
|
217 |
app.launch()
|