Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 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")
|