Update app.py

app.py

@@ -22,10 +22,13 @@ def func(text):
     with torch.no_grad():
         output = bert_sc(**encoding)
     scores = output.logits.argmax(-1)
+    neg = torch.softmax(output.logits, dim=1).tolist()[0][0]
+    pos = torch.softmax(output.logits, dim=1).tolist()[0][1]
     
     label = "ネガティブ" if scores.item()==0 else "ポジティブ"
+    cos = f"信頼度:{neg*100:.1f}%" if scores.item()==0 else f"信頼度:{pos*100:.1f}%"
 
-    return label
+    return label,cos
 
-app = gr.Interface(fn=func, inputs=gr.Textbox(lines=3, placeholder="文章を入力してください"), outputs="label", title="ビジネス文書のネガポジ分析", description=descriptions)
+app = gr.Interface(fn=func, inputs=gr.Textbox(lines=3, placeholder="文章を入力してください"), outputs=["label","label"], title="ビジネス文書のネガポジ分析", description=descriptions)
 app.launch()