Update app.py

app.py

@@ -28,7 +28,8 @@ def predict(img)->Tuple[Dict, float]:
     with torch.inference_mode():
         pred_probs=torch.softmax(effnetb2(img), dim=1)
 
-    pred_labels_and_probs={class_names[i]: float(pred_probs[0][i] for i in range(len(class_names)))}
+    pred_labels_and_probs={class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
+
     pred_time=round(timer()-start_time,5)
     return pred_labels_and_probs, pred_time