Update app.py

app.py

@@ -51,8 +51,8 @@ def classify(text):
         outputs = model(**inputs)
     probs = torch.softmax(outputs.logits, dim=1).numpy()[0]
     return {
-        "No": float(probs[0]),
-        "Yes": float(probs[1]),
+        "ไม่เกี่ยวข้อง": float(probs[0]),
+        "จ้างงานรถขนส่ง": float(probs[1]),
     }
 
 # gradio interface