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")
|