Update app.py

app.py

@@ -5,10 +5,14 @@ from transformers import AutoTokenizer
 #def greet(name):
 #    return "Hello " + name + "!!"
 
-def greet(sent):
-  
+def greet(sent,mode):
   print("input_sent= " + sent)
-  pt_model = 'best.pt'
+  if mode=='Malicious_comment':
+    pt_model ='best.pt'
+  if mode=='Economic_article':
+    pt_model ='best2.pt'
+  
+  
   print(pt_model)
   device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
   print("device:",device)
@@ -53,6 +57,7 @@ def greet(sent):
     result= sent + ">> 악의적인 내용이 보이지 않습니다."
     
   return result
-
-iface = gr.Interface(fn=greet,title='Korean malicious comment',description="한국어 악플 판별기",inputs="text", outputs="text")
+intput="text"
+input2= gr.Dropdown(choices=['Malicious_comment','Economic_article'])
+iface = gr.Interface(fn=greet,title='Korean malicious comment',description="한국어 악플 판별기",inputs=[intput,input2], outputs="text")
 iface.launch()