garrettbaber's picture
Update app.py
38d7836
raw history blame
No virus
522 Bytes
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()