Update app.py

app.py

@@ -24,6 +24,37 @@ word: """
 examples = [["river"], ["night"], ["trees"],["table"],["laughs"]]
 
 
+def poem2_generate(word):
+  p = word.lower() + "\n" + "poem using word: "
+  print(f"*****Inside poem_generate - Prompt is :{p}")
+  json_ = {"inputs": p,
+            "parameters":
+            {
+            "top_p": 0.9,
+          "temperature": 1.1,
+          "max_new_tokens": 50,
+          "return_full_text": False
+          }}
+  response = requests.post(API_URL, headers=headers, json=json_)
+  output = response.json()
+  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] # +"."
+  if "\n\n" not in output_tmp:
+    if output_tmp.find('.') != -1:
+      idx = output_tmp.find('.')
+      poem = output_tmp[:idx+1]
+    else:
+      idx = output_tmp.rfind('\n')
+      poem = output_tmp[:idx]
+  else:
+    poem = output_tmp.split("\n\n")[0] # +"."
+  poem = poem.replace('?','')
+  print(f"Poem being returned is: {poem}")
+  return poem
+  
+
 def poem_generate(word):
 
   p = prompt + word.lower() + "\n" + "poem using word: "
@@ -79,7 +110,7 @@ with demo:
   b1 = gr.Button("Generate Poem")
   b2 = gr.Button("Generate Image")
 
-  b1.click(poem_generate, input_word, poem_txt)
+  b1.click(poem2_generate, input_word, poem_txt)
   b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples