drod75 commited on
Commit
2973b77
·
verified ·
1 Parent(s): d06ab3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ pipe = pipeline(task="text-generation", model="google/gemma-2-2b-it")
14
 
15
  def poet(prompt):
16
  messages = [{"role": "user", "content": prompt}]
17
- outputs = pipe(messages, max_length=200)
18
  return outputs[0]["generated_text"][-1]["content"].strip()
19
 
20
  st.title("Shakespeare Ai")
 
14
 
15
  def poet(prompt):
16
  messages = [{"role": "user", "content": prompt}]
17
+ outputs = pipe(messages, max_length=80)
18
  return outputs[0]["generated_text"][-1]["content"].strip()
19
 
20
  st.title("Shakespeare Ai")