1

app.py

@@ -3,26 +3,26 @@ import requests
 
 # GPT-J-6B API
 API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
+prompt = """
+word: risk
+poem using word: And then the day came,
+when the risk
+to remain tight
+in a bud
+was more painful
+than the risk
+it took
+to blossom.
 
+word: """
 #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
-  poem using word: And then the day came,
-  when the risk
-  to remain tight
-  in a bud
-  was more painful
-  than the risk
-  it took
-  to blossom.
-  
-  word: """  
-  prompt = prompt + word.lower() + "\n" + "poem using word: "
-  print(f"Prompt is :{prompt}")
-  json_ = {"inputs": prompt,
+
+  p = prompt + word.lower() + "\n" + "poem using word: "
+  print(f"Prompt is :{p}")
+  json_ = {"inputs": p,
             "parameters":
             {
             "top_p": 0.9,
@@ -33,7 +33,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] # +"."
+  poem = output[0]['generated_text'].split("\n\n")[0] # +"."
   
   return poem