Update app.py

app.py

@@ -16,7 +16,7 @@ Never quite as it seems"""
 
 def poem2_generate(word):
   p = word.lower() + "\n" + "poem using word: "
-  print(f"*****Inside poem_generate - Prompt is :{p}")
+  gr.Markdown("f"*****Inside poem_generate - Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
             {
@@ -27,9 +27,9 @@ def poem2_generate(word):
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
-  print(f"If there was an error? Reason is : {output}")
+  gr.Markdown("f"If there was an error? Reason is : {output}")
   output_tmp = output[0]['generated_text']
-  print(f"GPTJ response without splits is: {output_tmp}")
+  gr.Markdown("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:
@@ -41,14 +41,14 @@ def poem2_generate(word):
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
   poem = poem.replace('?','')
-  print(f"Poem being returned is: {poem}")
+  gr.Markdown("f"Poem being returned is: {poem}")
   return poem
   
 
 def poem_generate(word):
 
   p = prompt + word.lower() + "\n" + "poem using word: "
-  print(f"*****Inside poem_generate - Prompt is :{p}")
+  gr.Markdown("f"*****Inside poem_generate - Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
             {
@@ -59,10 +59,10 @@ def poem_generate(word):
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
-  print(f"If there was an error? Reason is : {output}")
+  gr.Markdown("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] # +"."
+  gr.Markdown("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('.')
@@ -73,7 +73,7 @@ def poem_generate(word):
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
   poem = poem.replace('?','')
-  print(f"Poem being returned is: {poem}")
+  gr.Markdown("print(f"Poem being returned is: {poem}")
   return poem
 
 def poem_to_image(poem):