xinhe0616's picture
Create app.py
cc2a30b
raw
history blame
1.06 kB
from transformers import pipeline
model1 = pipeline("text-generation",model="uer/gpt2-chinese-poem")
from opencc import OpenCC
t2s=OpenCC('t2s').convert
s2t=OpenCC('s2t').convert
import jiagu
import gradio as gr
inp1=gr.inputs.Textbox(label="่ซ‹่ผธๅ…ฅไธ€ๆฎต่ฉฉ่ฉž")
inp2=gr.inputs.Slider(minimum=50,maximum=250,step=5,default=150, label="็”Ÿๆˆ่ฉฉ่ฉž้•ทๅบฆ")
op1=gr.outputs.Textbox(label="็”Ÿๆˆ่ฉฉ่ฉž็ตๆžœ")
op2=gr.outputs.Textbox(label="ๆƒ…ๆ„Ÿๅˆ†ๆž(positive=ๅๆญฃ้ข, negative=ๅ่ฒ ้ข)")
def poem_generator(txt,n=100):
model_input=t2s(txt)
poem=model1(model_input, max_length=n)[0]['generated_text']
poem="".join(poem.split())
sentiment = jiagu.sentiment(model_input)
smt=sentiment[0]
return s2t(poem),smt
iface=gr.Interface(fn=poem_generator,
inputs=[inp1,inp2],
outputs=[op1,op2],
title="็น้ซ”ไธญๆ–‡่ฉฉ่ฉž็”ŸๆˆWeb APP",
description="่ซ‹่ผธๅ…ฅไธ€ๆฎต่ฉฉ่ฉž๏ผŒๆˆ‘ๆœƒๅนซไฝ ็นผ็บŒ็”Ÿๆˆ่ฉฉ่ฉž๏ผŒ้‚„ๆœƒๅนซไฝ ๅˆ†ๆž้€™ๆ•ดๆฎต่ฉฉ่ฉžๅๆญฃ้ขๆˆ–่ฒ ้ขๅ–”!")
iface.launch()