Update app.py

app.py

@@ -28,7 +28,7 @@ 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)))}4
+    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