Update app.py

app.py

@@ -7,23 +7,22 @@ s2t=OpenCC('s2t').convert
 
 import jiagu
 
+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(poem)
+ return s2t(poem),sentiment[0]
+
 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="請輸入一段詩詞，我會幫你繼續生成詩詞，還會幫你分析這整段詩詞偏正面或負面喔!")
+                   description="請輸入一段文字，我會幫你繼續生成詩詞，還會幫你分析這整段詩詞是偏正面或負面喔!")
 iface.launch()