4 models

app.py

@@ -13,9 +13,17 @@ tokenizer = AutoTokenizer.from_pretrained("LazarusNLP/NusaBERT-large")
 bigru_model = BERTBiGRUClassifier.from_pretrained("Amal17/NusaBERT-concate-BiGRU-NusaX-ace")
 bigru_model.eval()
 
+bigru_translate_model = BERTBiGRUClassifier.from_pretrained("Amal17/NusaBERT-concate-BiGRU-NusaTranslate-senti")
+bigru_translate_model.eval()
+
 bilstm_model = BERTBiLSTMClassifier.from_pretrained("Amal17/NusaBERT-concate-BiLSTM-NusaX-ace")
 bilstm_model.eval()
 
+bilstm_translate_model = BERTBiLSTMClassifier.from_pretrained("Amal17/NusaBERT-concate-BiLSTM-NusaTranslate-senti")
+bilstm_translate_model.eval()
+
+
+
 # Inference helper
 def predict_with_model(model, text):
     inputs = tokenizer(
@@ -38,9 +46,14 @@ def compare_models(text):
     pred_a, conf_a = predict_with_model(bigru_model, text)
     pred_b, conf_b = predict_with_model(bilstm_model, text)
 
+    pred_c, conf_c = predict_with_model(bigru_translate_model, text)
+    pred_d, conf_d = predict_with_model(bilstm_translate_model, text)
+
     return (
-        f"BiGRU → Class: {pred_a}", f"Confidence: {conf_a:.4f}",
-        f"BiLSTM   → Class: {pred_b}", f"Confidence: {conf_b:.4f}"
+        f"Class: {pred_a} ({CLASS_MAP[pred_a]}) with confidence: {conf_a:.4f}",
+        f"Class: {pred_b} ({CLASS_MAP[pred_b]}) with confidence: {conf_b:.4f}",
+        f"Class: {pred_c} ({CLASS_MAP[pred_c]}) with confidence: {conf_c:.4f}",
+        f"Class: {pred_d} ({CLASS_MAP[pred_d]}) with confidence: {conf_d:.4f}",
     )
 
 # Build Gradio UI
@@ -48,12 +61,12 @@ interface = gr.Interface(
     fn=compare_models,
     inputs=gr.Textbox(label="Input Text"),
     outputs=[
-        gr.Textbox(label="BiGRU Prediction"),
-        gr.Textbox(label="BiGRU Confidence"),
-        gr.Textbox(label="BiLSTM Prediction"),
-        gr.Textbox(label="BiLSTM Confidence"),
+        gr.Textbox(label="NusaBERT-BiGRU-ace"),
+        gr.Textbox(label="NusaBERT-BiLSTM-ace"),
+        gr.Textbox(label="NusaBERT-BiGRU-translate"),
+        gr.Textbox(label="NusaBERT-BiLSRM-translate"),
     ],
-    title="Model Comparison: BiGRU vs BiLSTM"
+    title="Hybrid Model NusaBERT + RNN"
 )
 
 interface.launch()