Update app.py

app.py

@@ -9,12 +9,12 @@ clf = pipeline(model=model, task="image-classification", image_processor=image_p
 class_names = ['Benign', 'Malignant']
 
 def predict_image(img):
-  img=img.reshape(224,224,3)
   img = transforms.ToPILImage()(img)
+  img = transforms.Resize((224,224))(img)
   prediction=clf.predict(img)
   return {class_names[i]: float(prediction[i]["score"]) for i in range(2)}
 
-image = gr.Image(shape=(224,224))
+image = gr.Image(label="Select a mammogram image", sources=['upload'])
 label = gr.Label(num_top_classes=2)
 
-gr.Interface(fn=predict_image, inputs=image, outputs=label, interpretation='default', title="Mammogram classification").launch()
+gr.Interface(fn=predict_image, inputs=image, outputs=label, title="Mammogram classification").launch(debug='True')