Update app.py

app.py

@@ -46,13 +46,13 @@ def generate(prompt, history):
 
     for response in stream:
         output += response.token.text
-        yield [(prompt,output)],output
+        yield [(prompt,output)],output1
   
     if "PROMPT:" and "MEME_TEXT:" in output:
         print("YES")
-        prompt_t=output1.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0].strip()
+        prompt_t=output.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0].strip()
         print(prompt_t)
-        meme_t=output1.split("MEME_TEXT:",1)[1].strip()
+        meme_t=output.split("MEME_TEXT:",1)[1].strip()
         print(meme_t)
         output1={'PROMPT':prompt_t,'MEME_TEXT':meme_t}
         #output=str(output1)