Update app.py

app.py

@@ -49,15 +49,20 @@ generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOK
 
 def calculator(text1, operation, text2):
     if operation == "add":
-        return generator1(text1) + generator2(text2)
+        output = generator1(text1) + generator2(text2)
+        saved = AIMemory(text1 + " " + text2, output)
+        return output
     elif operation == "subtract":
         output = generator1(text1) + generator2(text2)
+        saved = AIMemory(text1 + " " + text2, output)
         return output.replace(text1, "").replace(text2, "")
     elif operation == "multiply":
         output = generator1(text1) + generator2(text2) + generator3(text1)
+        saved = AIMemory(text1 + " " + text2, output)
         return output
     elif operation == "divide":
         output = generator1(text2) + generator2(text1) + generator3(text2)
+        saved = AIMemory(text1 + " " + text2, output)
         return output.replace(text1, "").replace(text2, "")
 
 demo = gr.Interface(