Update app.py

app.py

@@ -58,7 +58,7 @@ def predict(img):
     z=Tensor(inp_img)
     y=z.view(-1,64)
     x=mdlV2(y)
-    w=F.softmax(x)
+    w=F.softmax(x,dim=0)
     v=w[0]
     u=v.data
     otpt=u