Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,12 +43,13 @@ client = InferenceClient(HF_MODEL_ID, token=os.getenv("HF_TOKEN"))
|
|
| 43 |
# return _llm
|
| 44 |
|
| 45 |
llm = HuggingFaceEndpoint(
|
| 46 |
-
repo_id="
|
|
|
|
| 47 |
task="text-generation",
|
| 48 |
max_new_tokens=512,
|
| 49 |
temperature=0.1,
|
| 50 |
huggingfacehub_api_token=os.getenv("HF_TOKEN"),
|
| 51 |
-
wait_for_model=True
|
| 52 |
)
|
| 53 |
|
| 54 |
|
|
|
|
| 43 |
# return _llm
|
| 44 |
|
| 45 |
llm = HuggingFaceEndpoint(
|
| 46 |
+
repo_id="meta-llama/Llama-3.1-8B-Instruct", # This is the "Live" 2026 model
|
| 47 |
+
# repo_id="ibm-granite/granite-3.1-8b-instruct", # Using Granite for 2026 stability
|
| 48 |
task="text-generation",
|
| 49 |
max_new_tokens=512,
|
| 50 |
temperature=0.1,
|
| 51 |
huggingfacehub_api_token=os.getenv("HF_TOKEN"),
|
| 52 |
+
wait_for_model=True
|
| 53 |
)
|
| 54 |
|
| 55 |
|