Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,15 @@ from transformers import (
|
|
16 |
# ================================
|
17 |
# 1️⃣ Authenticate with Hugging Face Hub
|
18 |
# ================================
|
19 |
-
|
20 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
# ================================
|
23 |
# 2️⃣ Load Model & Processor
|
|
|
16 |
# ================================
|
17 |
# 1️⃣ Authenticate with Hugging Face Hub
|
18 |
# ================================
|
19 |
+
|
20 |
+
# Get HF token securely from environment variables
|
21 |
+
HF_TOKEN = os.getenv("HF_TOKEN")
|
22 |
+
|
23 |
+
if HF_TOKEN is None:
|
24 |
+
raise ValueError("❌ Hugging Face API token not found. Please set it in Secrets.")
|
25 |
+
|
26 |
+
# Login using the stored token
|
27 |
+
login(token=HF_TOKEN)
|
28 |
|
29 |
# ================================
|
30 |
# 2️⃣ Load Model & Processor
|