category change

app.py

@@ -25,7 +25,7 @@ def predict(image_name):
     inputs = inputs.to(device)
     with torch.no_grad():
         outputs = model(inputs.unsqueeze(0))
-        values, indices = torch.topk(outputs, k=2) 
+        values, indices = torch.topk(outputs, k=3) 
         print(values,indices)
     return {i.item(): v.item() for i, v in zip(indices[0], values.detach()[0])}
 def preprocess(image_name):

weight.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:42b238af6dd26e746779eab9563b744a2f65275466753505a3785c9756782776
-size 44929133
+oid sha256:792c8a851a0d214580503f17406e89979149c5ab143ef140c6b41383613e5f27
+size 44931181