Update app.py

app.py

@@ -25,21 +25,21 @@ def predict_sentiment(text):
         return_tensors='tf'
     )
 
-    # ✅ Ubah pemanggilan model ke bentuk keyword arguments
-    outputs = model(
-        input_ids=inputs["input_ids"],
-        attention_mask=inputs["attention_mask"]
-    )
+    # Gabungkan input_ids dan attention_mask ke dalam 1 dict
+    model_inputs = {
+        "input_ids": inputs["input_ids"],
+        "attention_mask": inputs["attention_mask"]
+    }
+
+    outputs = model(inputs=model_inputs)
 
-    # Ambil hasil prediksi
+    # Ambil prediksi dari output dict
     prediction = tf.argmax(outputs["classifier"], axis=1).numpy()[0]
     return labels[prediction]
 
 # UI Gradio
-demo = gr.Interface(
-    fn=predict_sentiment,
-    inputs=gr.Textbox(label="Masukkan komentar"),
-    outputs=gr.Textbox(label="Hasil Sentimen")
-)
+demo = gr.Interface(fn=predict_sentiment,
+                    inputs=gr.Textbox(label="Masukkan komentar"),
+                    outputs=gr.Textbox(label="Hasil Sentimen"))
 
 demo.launch()