update display option

app.py

@@ -195,7 +195,7 @@ class inferSSCL():
         
         return s
     
-    def get_predict(self, top_pred, aspect_label, threshold=4):
+    def get_predict(self, top_pred, aspect_label, threshold=3):
         pred = {'none':0, 'do_an': 0, 'gia_ca':0, 'khong_gian': 0, 'phuc_vu': 0}
         try:
             for i in range(len(top_pred)):
@@ -274,4 +274,13 @@ else:
     outputs = model.output[0]
     if outputs:
         for output in outputs:
-            st.title(output)
+            if output == 'do_an':   
+                st.title('Đồ ăn')
+            elif output == 'gia_ca':
+                st.title('Giá cả')
+            elif output == 'khong_gian':
+                st.title('Không gian')
+            elif output == 'phuc_vu':
+                st.title('Phục vụ')
+            else:
+                st.title('None')