Spaces:
Paused
Paused
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()
|