Update app.py

app.py

@@ -1,11 +1,36 @@
 import gradio as gr
 import torch
 from nextus_regressor_class import *
+import nltk
 
 model = NextUsRegressor()
 model.load_state_dict(torch.load("./nextus_regressor1012.pt"))
 model.eval()
 
-gr.Interface(fn=model,
-            inputs=gr.Textbox(label="기사", lines=30, placeholder="기사를 입력하세요"),
-            outputs=gr.Number(label="Slant Score")).launch()
+def shap(txt, tok_level):
+    batch = [txt]
+    if tok_level == "word":
+        print("word")
+    elif tok_level == "sentence":
+        print("sentence")
+    else:
+        print("this token granularity not supported")
+    #tokens = nltk
+    with torch.no_grad():
+        y_pred = model(txt)
+    return y_pred
+
+demo = gr.Interface(fn=shap,
+             inputs=[
+                 gr.Textbox(label="기사", lines=30, placeholder="기사를 입력하세요."),
+                 gr.Radio(["sentence", "word"], value="sentence", info="문장 단위의 해설은 sentence를 단어 단위의 해설은 word를 선택하세요.")
+             ],
+            outputs=gr.HighlightedText(
+                label="Diff",
+                combine_adjacent=True,
+                show_legend=True,
+                color_map={"+": "red", "-": "green"}),
+                theme=gr.themes.Base())
+
+if __name__ == "__main__":
+    demo.launch()