1

app.py

@@ -60,7 +60,7 @@ def poem_generate(word):
             {
             "top_p": 0.9,
           "temperature": 1.1,
-          "max_new_tokens": 250,
+          "max_new_tokens": 500,
           "return_full_text": False
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
@@ -79,8 +79,8 @@ def poem_generate(word):
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
   poem = poem.replace('?','')
-  print(f"Poem being returned is: {poem}")
-  return poem
+  #print(f"Poem being returned is: {poem}")
+  return output_tmp #poem
 
 def poem_to_image(poem):
   print("*****Inside Poem_to_image")