Update app.py

app.py

@@ -25,7 +25,7 @@ def aiornot0(image):
         print (outputs)
         logits = outputs.logits
         print (logits)
-        probability = softmax(logits, axis=-1)
+        probability = softmax(logits)
         print(probability)
 
     prediction = logits.argmax(-1).item()