Spaces:
Building
Building
Update spark_startup.py
Browse files- spark_startup.py +12 -1
spark_startup.py
CHANGED
@@ -44,9 +44,20 @@ def notify_startup():
|
|
44 |
"fine_tune_zip": v.llm.fine_tune_zip,
|
45 |
}
|
46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
47 |
try:
|
48 |
log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
|
49 |
-
r = requests.post(f"{spark_base}/startup", json=body, timeout=10)
|
50 |
if r.status_code >= 400:
|
51 |
log(f"❌ Spark responded {r.status_code}: {r.text}")
|
52 |
r.raise_for_status()
|
|
|
44 |
"fine_tune_zip": v.llm.fine_tune_zip,
|
45 |
}
|
46 |
|
47 |
+
# Hugging Face token'ı ortam değişkeninden al
|
48 |
+
spark_token = os.getenv("SPARK_TOKEN")
|
49 |
+
if not spark_token:
|
50 |
+
log("❌ SPARK_TOKEN environment variable is missing.")
|
51 |
+
return
|
52 |
+
|
53 |
+
headers = {
|
54 |
+
"Authorization": f"Bearer {spark_token}",
|
55 |
+
"Content-Type": "application/json"
|
56 |
+
}
|
57 |
+
|
58 |
try:
|
59 |
log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
|
60 |
+
r = requests.post(f"{spark_base}/startup", json=body, headers=headers, timeout=10)
|
61 |
if r.status_code >= 400:
|
62 |
log(f"❌ Spark responded {r.status_code}: {r.text}")
|
63 |
r.raise_for_status()
|