drod75 commited on
Commit
d36b401
·
verified ·
1 Parent(s): 46ebd3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ pipe = pipeline(task="text-generation", model="google/gemma-2-2b-it")
15
  messages = []
16
 
17
  def poet(text):
18
- prompt = 'Make 25 lines of shakespeare based on: ' + text
19
  messages.append({"role": "user", "content": prompt})
20
  outputs = pipe(messages, max_length=80)
21
  return outputs[0]["generated_text"][-1]["content"].strip()
 
15
  messages = []
16
 
17
  def poet(text):
18
+ prompt = 'Make 25 lines, it has to be absolutely 25 lines of text no less no exception, of shakespeare based on this prompt: ' + text
19
  messages.append({"role": "user", "content": prompt})
20
  outputs = pipe(messages, max_length=80)
21
  return outputs[0]["generated_text"][-1]["content"].strip()