Spaces:
Sleeping
Sleeping
fix: Set max_new_tokens to 200
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
|
|
5 |
|
6 |
# Use a pipeline as a high-level helper
|
7 |
from transformers import pipeline
|
8 |
-
pipe = pipeline("text-generation", model="NousResearch/Hermes-3-Llama-3.1-8B")
|
9 |
|
10 |
@spaces.GPU(duration=120)
|
11 |
def llama3_1_8B(question):
|
|
|
5 |
|
6 |
# Use a pipeline as a high-level helper
|
7 |
from transformers import pipeline
|
8 |
+
pipe = pipeline("text-generation", model="NousResearch/Hermes-3-Llama-3.1-8B", max_new_tokens=200)
|
9 |
|
10 |
@spaces.GPU(duration=120)
|
11 |
def llama3_1_8B(question):
|