dd

app.py

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