Update app.py

app.py

@@ -23,26 +23,16 @@ def format_prompt(message, history):
 agents =[
     "MEME_GENERATOR",
 ]
-def generate(
-        prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
-):
+def generate(prompt, history):
     seed = random.randint(1,1111111111111111)
 
-    agent=MEME_GENERATOR
-    if agent_name == "MEME_GENERATOR":
-        agent = MEME_GENERATOR  
+    system_prompt=MEME_GENERATOR
   
-    system_prompt=agent
-    temperature = float(temperature)
-    if temperature < 1e-2:
-        temperature = 1e-2
-    top_p = float(top_p)
-
     generate_kwargs = dict(
-        temperature=temperature,
-        max_new_tokens=max_new_tokens,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
+        temperature=0.9,
+        max_new_tokens=256,
+        top_p=0.95,
+        repetition_penalty=1.0,
         do_sample=True,
         seed=seed,
     )
@@ -53,7 +43,7 @@ def generate(
 
     for response in stream:
         output += response.token.text
-        yield output
+        yield [(prompt,output)]
     with open('tmp.txt','w') as t:
         t.writelines(output)
     t.close()
@@ -70,7 +60,7 @@ def generate(
             except Exception as e:
                 print(e)
                 pass
-    return output
+    return [(prompt,output)]
     
 def run(inp,history,model_drop):
     #if len(inp)>max_prompt:
@@ -140,14 +130,16 @@ with gr.Blocks() as app:
             msg = gr.Textbox()
             model_drop=gr.Dropdown(label="Diffusion Models", type="index", choices=[m for m in models], value=models[0])
             with gr.Group():
-                submit_b = gr.Button()
+                submit_b = gr.Button("Meme")
+                submit_im = gr.Button("Image")
                 with gr.Row():
                     stop_b = gr.Button("Stop")
                     clear = gr.ClearButton([msg, chatbot])
         with gr.Column(scale=2):
             im_out=gr.Image(label="Image")
         
-    sub_b = submit_b.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
+    sub_b = submit_b.click(generate, [msg,chatbot],[chatbot])
+    sub_im = submit_b.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
     sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])
 app.launch()