1

app.py

@@ -40,7 +40,7 @@ def poem_generate(word):
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
-  print(f"Was there an error? Reason is : {output}")
+  print(f"If there was an error? Reason is : {output}")
   output_tmp = output[0]['generated_text']
   print(f"GPTJ response without splits is: {output_tmp}")
   #poem = output[0]['generated_text'].split("\n\n")[0] # +"."
@@ -53,6 +53,7 @@ def poem_generate(word):
       poem = output_tmp[:idx]
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
+  poem = poem.replace('?','')
   print(f"Poem being returned is: {poem}")
   return poem