Upload Lagi

app.py

@@ -20,13 +20,26 @@ def predict(text):
         pred_sentiment = sentiment_logits.argmax(dim=1).item()
         pred_lang = lang_logits.argmax(dim=1).item()
 
-    sentiment_label = "positive" if pred_sentiment == 1 else "negative"
-    lang_label = le.inverse_transform([pred_lang])[0]
+    if pred_sentiment == 2:
+        sentiment_label = "positive"
+    elif pred_sentiment == 1:
+        sentiment_label = "neutral"
+    else:
+        sentiment_label = "negative"
+
+    lang_code_map = {
+        'de': 'German',
+        'es': 'Espanyol',
+        'en': 'English',
+        'fr': 'French'
+    }
+        
+    lang_code = le.inverse_transform([pred_lang])[0]
+    lang_label = lang_code_map.get(lang_code, "Unknown")
 
     return sentiment_label, lang_label
 
 
-
 interface = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(label="Masukkan Teks Dalam Bahasa (Inggris/Belanda/Spanyol/Perancis)"),