Update app.py

app.py

@@ -67,9 +67,13 @@ def generate(prompt, history):
     output1=output.replace("\n"," ")
     if "PROMPT:" and "MEME_TEXT:" in output1:
         print("YES")
-        output1={'PROMPT':output1.split("PROMPT:",1)[1].split("MEME_TEXT",1)[0],'MEME_TEXT':output1.split("MEME_TEXT",1)[1]}
+        prompt_t=output1.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0]
+        print(prompt_t)
+        meme_t=output1.split("MEME_TEXT",1)[1]
+        print(meme_t)
+        output1={'PROMPT':prompt_t,'MEME_TEXT':meme_t}
         output=str(output1)
-    return [(prompt,output)],output1
+    yield [(prompt,output)],output1
     
 def run(inp,history,model_drop):
     #if len(inp)>max_prompt: