Spaces:
Paused
Paused
Ali Mohsin
commited on
Commit
·
ec83d0c
1
Parent(s):
dce7a6a
more fixes
Browse files
app.py
CHANGED
|
@@ -5,7 +5,8 @@ import random
|
|
| 5 |
import os
|
| 6 |
|
| 7 |
# Initialize the client (uses HUGGING_FACE_HUB_TOKEN from environment)
|
| 8 |
-
|
|
|
|
| 9 |
|
| 10 |
SYSTEM_PROMPT = """You are ModelForge, an expert AI architecture assistant. Your goal is to analyze machine learning problems and generate detailed, deployable solutions in strict JSON format.
|
| 11 |
|
|
|
|
| 5 |
import os
|
| 6 |
|
| 7 |
# Initialize the client (uses HUGGING_FACE_HUB_TOKEN from environment)
|
| 8 |
+
token = os.getenv("HUGGING_FACE_HUB_TOKEN")
|
| 9 |
+
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=token)
|
| 10 |
|
| 11 |
SYSTEM_PROMPT = """You are ModelForge, an expert AI architecture assistant. Your goal is to analyze machine learning problems and generate detailed, deployable solutions in strict JSON format.
|
| 12 |
|