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