Create app.py

app.py

@@ -0,0 +1,29 @@
+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()