File size: 740 Bytes
748b73a
dc772a7
 
 
748b73a
dc772a7
 
748b73a
dc772a7
e9b8265
dc772a7
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import gradio as gr
from transformers import BertForSequenceClassification
from transformers import BertTokenizer
import torch

tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-330M-Sentiment')
model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-330M-Sentiment')

def predict(input_text):
    predictions = model(torch.tensor([tokenizer.encode(input_text)]))
    predictions=torch.nn.functional.softmax(predictions.logits,dim=-1)
    return input_text, {p["label"]: p["score"] for p in predictions} 

gradio_app = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
    title="ζˆ‘θƒ½εšδΈ­ζ–‡ζ–‡ζœ¬ηš„ζƒ…η»ͺδΊŒεˆ†η±»",
)

if __name__ == "__main__":
    gradio_app.launch()