Update app.py

app.py

@@ -2,11 +2,11 @@ import gc
 import gradio as gr
 from transformers import pipeline
 
-pipe = pipeline('text-generation', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
+pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
 #gc.collect()
 
 def sayPoetry(prompt):
-    gen = pipe(prompt, max_length=96)[0]["generated_text"]
+    gen = pipe(prompt, max_length=96, temperature = 0.95)[0]["generated_text"]
     poetry =""
     for line in gen.split('.')[:-1]:
         poetry += line