Update app.py

app.py

@@ -18,7 +18,7 @@ def predict_image(img):
   img = transforms.Resize((256,256))(img)
   prediction=clf.predict(img)
   print(prediction)
-  print(model.config.id2label[predicted_label])
+
   return {class_names[i]: float(prediction[i]["score"]) for i in range(2)}
 
 image = gr.Image(label="Image to Analyze", sources=['upload'])