test

app.py

@@ -53,7 +53,7 @@ def predict(inp):
   with torch.no_grad():
     prediction = model(img).softmax(1).numpy()
     confidences = {labels[i]: float(prediction[0][i]) for i in range(2)}    
-  return confidences, img
+  return confidences, inp
 
 import gradio as gr