Update app.py

app.py

@@ -24,21 +24,22 @@ def shap(txt, tok_level):
     with torch.no_grad():
         y_pred = model(txt)
         y_offs = model(batch)
-        shaps = (y_offs - y_pred).tolist()[0] # convert to list and make tuple to be returned
-    labels = ["+" if s < 0 else "-" for s in shaps]
-    return str(list(zip(tokens, labels)))
+        shaps = (y_offs - y_pred).tolist() # convert to list and make tuple to be returned
+    shapss = [s[0] for s in shaps]
+    labels = ["+" if s < 0 else "-" for s in shapss]
+    return list(zip(tokens, labels))
 
 demo = gr.Interface(shap,
                     [
                          gr.Textbox(label="기사", lines=30, placeholder="기사를 입력하세요."),
                          gr.Radio(choices=["sentence", "word"], label="해설 표시 단위", value="sentence", info="문장 단위의 해설은 sentence를, 단어 단위의 해설은 word를 선택하세요.")
                     ],
-                    gr.Textbox(label="Test Output"),
-                    #gr.HighlightedText(
-                    #    label="Diff",
-                    #    combine_adjacent=True,
-                    #    show_legend=True,
-                    #    color_map={"+": "red", "-": "green"}),
+                    #gr.Textbox(label="Test Output"),
+                    gr.HighlightedText(
+                        label="Diff",
+                        combine_adjacent=True,
+                        show_legend=True,
+                        color_map={"+": "red", "-": "green"}),
                    
                     theme=gr.themes.Base())