inital
This commit is contained in:
commit
16917ed119
|
@ -0,0 +1,51 @@
|
||||||
|
import gradio as gr
|
||||||
|
from transformers import AutoModelForSequenceClassification
|
||||||
|
from transformers import TFAutoModelForSequenceClassification
|
||||||
|
from transformers import AutoTokenizer, AutoConfig
|
||||||
|
import numpy as np
|
||||||
|
from scipy.special import softmax
|
||||||
|
# Preprocess text (username and link placeholders)
|
||||||
|
def preprocess(text):
|
||||||
|
new_text = []
|
||||||
|
for t in text.split(" "):
|
||||||
|
t = '@user' if t.startswith('@') and len(t) > 1 else t
|
||||||
|
t = 'http' if t.startswith('http') else t
|
||||||
|
new_text.append(t)
|
||||||
|
return " ".join(new_text)
|
||||||
|
|
||||||
|
def inference(text):
|
||||||
|
model_path="twitter-roberta-base-sentiment-latest"
|
||||||
|
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
||||||
|
config = AutoConfig.from_pretrained(model_path)
|
||||||
|
# PT
|
||||||
|
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
||||||
|
|
||||||
|
text = preprocess(text)
|
||||||
|
encoded_input = tokenizer(text, return_tensors='pt')
|
||||||
|
output = model(**encoded_input)
|
||||||
|
scores = output[0][0].detach().numpy()
|
||||||
|
scores = softmax(scores)
|
||||||
|
|
||||||
|
ranking = np.argsort(scores)
|
||||||
|
ranking = ranking[::-1]
|
||||||
|
for i in range(scores.shape[0]):
|
||||||
|
l = config.id2label[ranking[i]]
|
||||||
|
s = scores[ranking[i]]
|
||||||
|
return f"{i+1}) {l} {np.round(float(s), 4)}"
|
||||||
|
|
||||||
|
|
||||||
|
title = "sentiment analysis:twitter-roberta-base-sentiment"
|
||||||
|
description = "Gradio Demo for twitter-roberta-base-sentiment. To use it, simply upload your image, or click one of the examples to load them."
|
||||||
|
article = "<p style='text-align: center'><a href='https://github.com/bryandlee/animegan2-pytorch' target='_blank'>Github Repo Pytorch</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_animegan' alt='visitor badge'></center></p>"
|
||||||
|
examples=[["Covid cases are increasing fast!"]]
|
||||||
|
|
||||||
|
demo = gr.Interface(
|
||||||
|
fn=inference,
|
||||||
|
inputs=[gr.inputs.Textbox()],
|
||||||
|
outputs=gr.outputs.Textbox(),
|
||||||
|
title=title,
|
||||||
|
description=description,
|
||||||
|
article=article,
|
||||||
|
examples=examples)
|
||||||
|
|
||||||
|
demo.launch()
|
|
@ -0,0 +1 @@
|
||||||
|
Subproject commit f8f82141e1c60fb0869fc353c1e84a944d4f920c
|
Loading…
Reference in New Issue