1

app.py

@@ -12,27 +12,6 @@ HF_TOKEN = os.environ["HF_TOKEN"]
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 
-prompt1 = """
-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: """
-
-prompt2 = """
-Q: Joy has 5 balls. He buys 2 more cans of balls. Each can has 3 balls. How many balls he has now?
-A: Joy had 5 balls. 2 cans of 3 balls each is 6 balls. 5 + 6 = 11. Answer is 11.
-Q: Jane has 16 balls. Half balls are golf balls, and half golf balls are red. How many red golf balls are there?
-A: """
-
-prompt3 = """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?
-A: Let’s think step by step.
-"""
 #Complete below sentence in fun way.
 prompt4 = """Distracted from: hubble
 by: james webb
@@ -96,7 +75,7 @@ def write_on_image(final_solution):
   
   return image0, new_prompt
 
-def meme_generate(img, prompt): #prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
+def meme_generate(img, prompt, temp, top_p): #prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
 
   print(f"*****Inside meme_generate - Prompt is :{prompt}")
   if len(prompt) == 0:
@@ -106,10 +85,10 @@ def meme_generate(img, prompt): #prompt, generated_txt): #, input_prompt_sql ):
             "parameters":
             {
           #"top_p": 0.95, 
-          "top_p": 0.90,
+          "top_p": top_p, #0.90,
           #"top_k":0,
           "max_new_tokens": 250,
-          "temperature": 1.1,
+          "temperature": temp, #1.1,
           #"num_return_sequences": 3,
           "return_full_text": True,
           "do_sample": True,
@@ -126,14 +105,6 @@ def meme_generate(img, prompt): #prompt, generated_txt): #, input_prompt_sql ):
   print(f"output_tmp is: {output_tmp}")
   solution = output_tmp.split("\nQ:")[0]   
   print(f"Final response after splits is: {solution}")
-  #if '\nOutput:' in solution:
-  #  final_solution = solution.split("\nOutput:")[0] 
-  #  print(f"Response after removing output is: {final_solution}")
-  #elif '\n\n' in solution:
-  #  final_solution = solution.split("\n\n")[0] 
-  #  print(f"Response after removing new line entries is: {final_solution}")
-  #else:
-  #final_solution = solution
     
   meme_image, new_prompt = write_on_image(solution)  
   return meme_image, new_prompt #final_solution #display_output, new_prompt  #generated_txt+prompt
@@ -168,12 +139,16 @@ with demo:
 
   #with gr.Row():
     output_image = gr.Image() #type="filepath", shape=(256,256))
+  with gr.Row():
+    in_slider_temp = gr.Slider(minimum=0.0, maximum=2.0, value=1.1, step=0.1, label='Temperature')
+    in_slider_top_p = gr.Slider(minimum=0.50, maximum=0.99, value=0.90, step=0.01, label='Top_p')
+  
     #output_prompt = gr.Textbox(label="Text generated", lines=5)
     
   b1 = gr.Button("Generate")
   #b2 = gr.Button("Generate Image") 
 
-  b1.click(meme_generate, inputs=[in_image, input_prompt] , outputs=[output_image,input_prompt])  #output_prompt  
+  b1.click(meme_generate, inputs=[in_image, input_prompt, in_slider_temp, in_slider_top_p] , outputs=[output_image,input_prompt])  #output_prompt  
   #b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples