import gradio as gr
import os
from setfit import SetFitModel


def greet(name):
    return "Hello " + name + "!!"

model = SetFitModel.from_pretrained("aalkaswan/java-deprecation-classifier", 
                                    device='cpu', #Use this if you don't have a GPU 
                                    token=os.environ.get('sec_tok'))

def classify(text):
    return model(text)
    
iface = gr.Interface(fn=classify, inputs="text", outputs="text")
iface.launch()