1

app.py

@@ -4,11 +4,12 @@ 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 "]]
+#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):
-  prompt = """word: risk
+  prompt = """
+  word: risk
   poem using word: And then the day came,
   when the risk
   to remain tight
@@ -18,9 +19,9 @@ def poem_generate(word):
   it took
   to blossom.
   
-  word: """
-  
+  word: """  
   prompt = prompt + word.lower() + "\n" + "poem using word: "
+  print(f"Prompt is :{prompt}")
   json_ = {"inputs": prompt,
             "parameters":
             {
@@ -31,6 +32,7 @@ def poem_generate(word):
           }}
   response = requests.post(API_URL, json=json_)
   output = response.json()
+  print(f"Poem without splits is: {output[0]['generated_text']}")
   poem = output[0]['generated_text'].split("\n\n")[0] # +"."
   
   return poem