Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,8 +46,8 @@ login(token=hf_token)
|
|
| 46 |
model_pipeline = pipeline(
|
| 47 |
model="meta-llama/Llama-3.2-1B",
|
| 48 |
#use_auth_token=hf_token,
|
| 49 |
-
max_length=
|
| 50 |
-
max_new_tokens=
|
| 51 |
)
|
| 52 |
|
| 53 |
# Use the pipeline in HuggingFacePipeline
|
|
|
|
| 46 |
model_pipeline = pipeline(
|
| 47 |
model="meta-llama/Llama-3.2-1B",
|
| 48 |
#use_auth_token=hf_token,
|
| 49 |
+
max_length=1000, # You can increase this if needed
|
| 50 |
+
max_new_tokens=500 # Limit how many tokens are generated
|
| 51 |
)
|
| 52 |
|
| 53 |
# Use the pipeline in HuggingFacePipeline
|