Update app.py

app.py

@@ -24,7 +24,7 @@ def poet(text):
     chat = [{"role": "user", "content": prompt}]
     prompt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
     inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
-    outputs = model.generate(input_ids=inputs.to(model.device), max_new_tokens=150)
+    outputs = model.generate(input_ids=inputs.to(model.device), max_new_tokens=350)
     return tokenizer.decode(outputs[0])
     
 st.title("Shakespeare Ai")