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: "
-  gr.Markdown("f*****Inside poem_generate - Prompt is :{p}")
+  gr.Markdown("Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
             {
@@ -27,10 +27,10 @@ def poem2_generate(word):
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
-  gr.Markdown("f"If there was an error? Reason is : {output}")
+  gr.Markdown("error? Reason is : {output}")
   output_tmp = output[0]['generated_text']
-  gr.Markdown("f"GPTJ response without splits is: {output_tmp}")
-  #poem = output[0]['generated_text'].split("\n\n")[0] # +"."
+  gr.Markdown("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('.')
@@ -41,14 +41,14 @@ def poem2_generate(word):
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
   poem = poem.replace('?','')
-  gr.Markdown("f"Poem being returned is: {poem}")
+  gr.Markdown("Returned is: {poem}")
   return poem
   
 
 def poem_generate(word):
 
   p = prompt + word.lower() + "\n" + "poem using word: "
-  gr.Markdown("f"*****Inside poem_generate - Prompt is :{p}")
+  gr.Markdown("Generate - Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
             {
@@ -59,9 +59,9 @@ def poem_generate(word):
           }}
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
-  gr.Markdown("f"If there was an error? Reason is : {output}")
+  gr.Markdown(error? Reason is : {output}")
   output_tmp = output[0]['generated_text']
-  gr.Markdown("f"GPTJ response without splits is: {output_tmp}")
+  gr.Markdown("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:
@@ -73,11 +73,11 @@ def poem_generate(word):
   else:
     poem = output_tmp.split("\n\n")[0] # +"."
   poem = poem.replace('?','')
-  gr.Markdown("print(f"Poem being returned is: {poem}")
+  gr.Markdown("Returned is: {poem}")
   return poem
 
 def poem_to_image(poem):
-  gr.Markdown("*****Inside Poem_to_image")
+  gr.Markdown("toimage")
   poem = " ".join(poem.split('\n'))
   poem = poem + " oil on canvas."
   steps, width, height, images, diversity = '50','256','256','1',15
@@ -91,10 +91,8 @@ def set_example(example: list) -> dict:
 demo = gr.Blocks()
 
 with demo:
-  gr.Markdown("<h1><center>Few Shot Learning for Text - Word Image Search</center></h1>")
-  gr.Markdown(
-        "https://huggingface.co/blog/few-shot-learning-gpt-neo-and-inference-api,  https://github.com/EleutherAI/the-pile"
-    )
+  gr.Markdown("<h1><center>Few Shot Learning Text to Word Image Search</center></h1>")
+  gr.Markdown("https://huggingface.co/blog/few-shot-learning-gpt-neo-and-inference-api,  https://github.com/EleutherAI/the-pile")
   with gr.Row():
     input_word = gr.Textbox(lines=7, value=prompt)