Update app.py

app.py

@@ -50,11 +50,20 @@ def greet(sent,mode):
   logits = outputs[0]   ## 마지막 노드에서 아무런 Activation Function을 거치지 않은 값을 Logit
   logits = logits.detach().cpu()
   result = logits.argmax(-1)
-  if result == 0:
-    result = sent + ">> 악성글로 판단됩니다. 조심하세요."
-    
-  elif result ==1:
-    result= sent + ">> 악의적인 내용이 보이지 않습니다."
+  if mode=='Malicious_comment':
+    if result == 0:
+      result = sent + ">> 악성글로 판단됩니다. 조심하세요."
+      
+    elif result ==1:
+      result= sent + ">> 악의적인 내용이 보이지 않습니다."
+  elif mode=='Economic_article':
+    if result == 0:
+      result = "중립"
+    elif result == 1:
+      result = "긍정"
+    elif result == 2:
+      result = "부정"
+    return result
     
   return result
 intput="text"