update

app.py

@@ -4,6 +4,7 @@ import requests
 # GPT-J-6B API
 API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
 
+examples = [["the roads are lovely dark and deep"], ["to be or not to be is the question, isn't it"], ["Elon Musk is not sure about "]]
 
 
 def poem_generate(word):
@@ -20,7 +21,7 @@ def poem_generate(word):
   
   word: """
   
-  prompt = prompt + word + "\n" + "poem using word: "
+  prompt = prompt + word.lower() + "\n" + "poem using word: "
   json_ = {"inputs": prompt,
             "parameters":
             {
@@ -45,6 +46,11 @@ def poem_to_image(poem):
 demo = gr.Blocks()
 
 with demo:
+  gr.Markdown("<h1><center>Generate Short Poem along with an Illustration</center></h1>")
+  gr.Markdown(
+        "<div>Enter a single word you would want GPTJ-6B to write poetry on.</div>"
+        "<div>Generate an illustration provided by Latent Diffusion model.</div><div>GPJ-6B is a 6 Billion parameter autoregressive language model. It generates the Poem based on how it has been 'prompt-engineered' 🤗 The complete text of generated poem then goes in as a prompt to the amazing Latent Diffusion Art space by <a href='https://huggingface.co/spaces/multimodalart/latentdiffusion' target='_blank'>Multimodalart</a>.</div>"
+    )
   input_word = gr.Textbox(placeholder="Enter a word here to create a Poem on..")
   poem_txt = gr.Textbox(lines=7)
   output_image = gr.Image(type="filepath", shape=(256,256))