Mikoranks / app.Py
Shinhati2023's picture
Update app.Py
fa4564e verified
raw
history blame
No virus
566 Bytes
import gradio as gr
from transformers import pipeline
calculator = pipeline("zero-shot-classification")
def calculate_expression(expression):
result = calculator(expression, candidate_labels=["addition", "subtraction", "multiplication", "division"])
if result["labels"][0] == "addition":
return eval(expression)
elif result["labels"][0] == "subtraction":
return eval(expression)
elif result["labels"][0] == "multiplication":
return eval(expression)
elif result["labels"][0] == "division":
return eval(expression)