Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ class BasicAgent:
|
|
27 |
|
28 |
# Initialize the model
|
29 |
model = HfApiModel(model_id="meta-llama/Llama-3.1-8B-Instruct",
|
30 |
-
format="text-generation",
|
31 |
token=os.environ["HF_TOKEN"],
|
32 |
max_tokens=2048,
|
33 |
temperature=0.0
|
|
|
27 |
|
28 |
# Initialize the model
|
29 |
model = HfApiModel(model_id="meta-llama/Llama-3.1-8B-Instruct",
|
30 |
+
# format="text-generation",
|
31 |
token=os.environ["HF_TOKEN"],
|
32 |
max_tokens=2048,
|
33 |
temperature=0.0
|