File size: 566 Bytes
e8fd854
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)