Fix label outputs with Japanese names

app.py

@@ -27,6 +27,12 @@ def load_model():
     model.load_state_dict(filtered_state, strict=False)
     model.eval()
     device = 0 if torch.cuda.is_available() else -1
+    model.config.id2label = {
+        0: "うどん",
+        1: "蕎麦",
+        2: "どちらでもない",
+    }
+    model.config.label2id = {v: k for k, v in model.config.id2label.items()}
     pipe = pipeline(
         "image-classification",
         model=model,