AhmadT198 commited on
Commit
59cdc11
1 Parent(s): 522fd1b

fix: Set max_new_tokens to 200

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):