update app.py

app.py

@@ -16,16 +16,6 @@ headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 #})
 
 
-# GPT-J-6B API
-#API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
-#HF_TOKEN = os.environ["HF_TOKEN"]
-#headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-#word: bird
-#poem using word: She sights a bird, she chuckles
-#She flattens, then she crawls
-#She runs without the look of feet
-#Her eyes increase to Balls.
-
 prompt1 = """
 word: risk
 poem using word: And then the day came,
@@ -48,12 +38,10 @@ prompt3 = """Q: A juggler can juggle 16 balls. Half of the balls are golf balls,
 A: Let’s think step by step.
 """
 
-examples = [["river"], ["night"], ["trees"],["table"],["laughs"]]
 
 
 def text_generate(prompt):
 
-  #p = prompt .lower() + "\n" + "poem using word: "
   print(f"*****Inside poem_generate - Prompt is :{prompt}")
   json_ = {"inputs": prompt,
             "parameters":
@@ -69,7 +57,7 @@ def text_generate(prompt):
   print(f"output is : {output}") #{output}")
   output_tmp = output[0]['generated_text']
   print(f"Bloom response without splits is: {output_tmp}")
-  solution = output[0]['generated_text'].split("\n\n")[0] # +"."
+  solution = output[0]['generated_text'].split("\nQ:")[0] # +"."
   print(f"Final response after splits is: {solution}")
   #if "\n\n" not in output_tmp:
   #  if output_tmp.find('.') != -1:
@@ -82,7 +70,7 @@ def text_generate(prompt):
   #  poem = output_tmp.split("\n\n")[0] # +"."
   #poem = poem.replace('?','')
   #print(f"Poem being returned is: {poem}")
-  return output #response #output_tmp #poem
+  return solution #output #response #output_tmp #poem
 
 
 demo = gr.Blocks()
@@ -93,7 +81,7 @@ with demo:
         """Testing Bloom """
         )
   with gr.Row():
-    example_prompt = gr.Radio( ["Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there? \nA: Let’s think step by step.\n"], label= "Choose a sample Prompt")
+    example_prompt = gr.Radio( ["Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA: Let’s think step by step.\n"], label= "Choose a sample Prompt")
     #input_word = gr.Textbox(placeholder="Enter a word here to generate text ...")
     generated_txt = gr.Textbox(lines=7)
     #output_image = gr.Image(type="filepath", shape=(256,256))