Update app.py

app.py

@@ -67,19 +67,20 @@ def generate(prompt, history):
     output1=output.replace("\n"," ")
     if "PROMPT:" and "MEME_TEXT:" in output1:
         print("YES")
-        prompt_t=output1.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0]
+        prompt_t=output1.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0].strip()
         print(prompt_t)
-        meme_t=output1.split("MEME_TEXT",1)[1]
+        meme_t=output1.split("MEME_TEXT:",1)[1].strip()
         print(meme_t)
         output1={'PROMPT':prompt_t,'MEME_TEXT':meme_t}
-        output=str(output1)
+        #output=str(output1)
     yield [(prompt,output)],output1
     
-def run(inp,history,model_drop):
+def run(inp,model_drop):
+    prompt=inp['PROMPT']
     #if len(inp)>max_prompt:
     #    inp = condense(inp)
     print(f'history :: {history}')
-    prompt=format_prompt(inp,history)
+    #prompt=format_prompt(inp,history)
     seed = random.randint(1,1111111111111111)
     print (seed)
     generate_kwargs = dict(
@@ -153,7 +154,7 @@ with gr.Blocks() as app:
     json_out=gr.JSON()
         
     sub_b = submit_b.click(generate, [msg,chatbot],[chatbot,json_out])
-    sub_im = submit_im.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
+    sub_im = submit_im.click(run, [json_out,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()