Update app.py

app.py

@@ -27,7 +27,7 @@ try:
 except:
     print("file not found")
     
-def AIMemory(name: str, message: str, context: str ):
+def AIMemory(name: str, message: str):
     if name and message:
         with open(DATA_FILE, "a") as csvfile:
             writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
@@ -47,7 +47,7 @@ generator1 = gr.Interface.load("huggingface/gpt2-large", api_key=HF_TOKEN)
 generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
 generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
 
-def calculator(text1, operation, text2):
+def calculator(text1, operation, text2, context):
     if operation == "add":
         output = generator1(text1) + generator2(text2)
         saved = AIMemory(text1 + " " + text2, output)