Update app.py

app.py

@@ -1,16 +1,23 @@
 import gradio as gr
 
-from transformers import pipeline
-
-pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
-
-def predict(text):
-  return pipe(text)[0]["translation_text"]
+def calculator(num1, operation, num2):
+    if operation == "add":
+        return num1 + num2
+    elif operation == "subtract":
+        return num1 - num2
+    elif operation == "multiply":
+        return num1 * num2
+    elif operation == "divide":
+        return num1 / num2
 
 demo = gr.Interface(
-  fn=predict,
-  inputs='text',
-  outputs='text',
+    calculator,
+    [
+        "number",
+        gr.Radio(["add", "subtract", "multiply", "divide"]),
+        "number"
+    ],
+    "number",
+    live=True,
 )
-
 demo.launch()