Update app.py

app.py

@@ -38,6 +38,7 @@ def aiornot(image,mod_choose):
       logits = outputs.logits
       print (logits)
     prediction = logits.argmax(-1).item()
+    print(prediction)
     label = labels[prediction] 
     return label