app.py

import gradio as gr
import os
from setfit import SetFitModel


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

# model = SetFitModel.from_pretrained("AISE-TUDelft/java-ownership-classifier", use_auth_token=os.environ.get('sec_tok'))
model = SetFitModel.from_pretrained("dvilasuero/setfit-mini-imdb")

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