Update app.py

app.py

@@ -26,7 +26,7 @@ def predict_single_text(text):
     with torch.no_grad():
         logits = model(**inputs).logits
     probs = torch.softmax(logits, dim=1)[0].tolist()
-    return {TARGET_NAMES[i]: v for i, v in enumerate(probs)}
+    return {target_names[i]: v for i, v in enumerate(probs)}
 
 def process_csv_file(file_obj):
     """The enterprise logic: CSV -> Classified CSV"""