Update app.py

app.py

@@ -45,23 +45,29 @@ 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):
-    if operation == "add":
-        output = generator1(text1) + generator2(text2)
-        saved = AIMemory(text1 + " " + text2, output)
+def calculator(intro, operator, outro):
+    if operator == "add":
+        output = generator2(intro) + generator3(outro)
+        title = intro + " " + outro
+        saved = AIMemory(title, output)
         return output
-    elif operation == "subtract":
-        output = generator1(text2) + generator2(text1)
-        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)
+    elif operator == "subtract":
+        output = generator2(outro) + generator3(intro)
+        title = outro + " " + intro
+        saved = AIMemory(title, output)
+        output = output.replace(intro, "").replace(outro, "")
+        return output
+    elif operator == "multiply":
+        output = generator1(intro) + generator2(outro) + generator3(intro)
+        title = intro + " " + outro + " " + intro
+        saved = AIMemory(title, output)
+        return output
+    elif operator == "divide":
+        output = generator1(outro) + generator2(intro) + generator3(outro)
+        title = outro + " " + intro + " " + outro
+        saved = AIMemory(title, output)
+        output = output.replace(intro, "").replace(outro, "")
         return output
-    elif operation == "divide":
-        output = generator1(text2) + generator2(text1) + generator3(text2)
-        saved = AIMemory(text1 + " " + text2, output)
-        return output.replace(text1, "").replace(text2, "")
 
 #with open('Mindfulness.txt', 'r') as file:
 #    context = file.read()
@@ -91,6 +97,7 @@ demo = gr.Interface(
     ],
     "text",
     examples=examples,
+    article="Saved story memory is at URL: https://huggingface.co/datasets/awacke1/MindfulStory.csv",
     live=True,
 )
 demo.launch()