import gradio as gr | |
from transformers import AutoModelForSequenceClassification, AutoTokenizer | |
def processInput(data): | |
model = AutoModelForSequenceClassification.from_pretrained("garrettbaber/twitter-roberta-base-fear-intensity") | |
tokenizer = AutoTokenizer.from_pretrained("garrettbaber/twitter-roberta-base-fear-intensity") | |
inputs = tokenizer("I'm scared", return_tensors="pt") | |
outputs = model(**inputs) | |
return outputs | |
app = gr.Interface(fn=processInput, inputs="text", outputs="text") | |
app.launch() |