update best label for model

app.py

@@ -45,7 +45,7 @@ with row1_1:
 with row1_2:
     selected_model = st.selectbox(
         "Choose the model",
-        ("IndoBERT", "IndoBERT-CNN (Best)", "IndoBERT-LSTM"),
+        ("IndoBERT", "IndoBERT-CNN", "IndoBERT-LSTM (Best)"),
     )
 
 # text input
@@ -185,7 +185,7 @@ def show_result(outputs):
 
     # create note if wanna download, hove on top right in table show
     st.markdown("**Note:** To download the table, hover over the top right corner of the table and click the download button.")
-    
+
 
 
 

model_services/pipeline.py

@@ -65,9 +65,9 @@ def predict(text: str, model_name: str):
     model = None
     if model_name == "IndoBERT":
         model = indobert_model
-    if model_name == "IndoBERT-CNN (Best)":
+    if model_name == "IndoBERT-CNN":
         model = indobertcnn_model
-    if model_name == "IndoBERT-LSTM":
+    if model_name == "IndoBERT-LSTM (Best)":
         model = indobertlstm_model
     input_ids = token_result['input_ids']
     attention_mask = token_result['attention_mask']