Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,8 @@ class BasicAgent:
|
|
| 21 |
def __init__(self, model_id: str = "HuggingFaceH4/zephyr-7b-beta"):
|
| 22 |
model_id_env = os.getenv("HF_MODEL_ID")
|
| 23 |
self.model_id = model_id_env.strip() if model_id_env else model_id
|
| 24 |
-
|
|
|
|
| 25 |
self.api_token = os.getenv("HF_TOKEN")
|
| 26 |
if not self.api_token:
|
| 27 |
raise ValueError("❌ Hugging Face API token not found. Please set 'HF_TOKEN' as env variable.")
|
|
|
|
| 21 |
def __init__(self, model_id: str = "HuggingFaceH4/zephyr-7b-beta"):
|
| 22 |
model_id_env = os.getenv("HF_MODEL_ID")
|
| 23 |
self.model_id = model_id_env.strip() if model_id_env else model_id
|
| 24 |
+
|
| 25 |
+
self.api_url = f"https://huggingface.co/api/models/{self.model_id}" #https://api-inference.huggingface.co/models/
|
| 26 |
self.api_token = os.getenv("HF_TOKEN")
|
| 27 |
if not self.api_token:
|
| 28 |
raise ValueError("❌ Hugging Face API token not found. Please set 'HF_TOKEN' as env variable.")
|