Update app.py

app.py

@@ -20,7 +20,7 @@ labels = ['Acne / Rosacea',
                   'Urticaria Hives', 'Vascular Tumors', 'Vasculitis', 'Warts Molluscum']
 
 def classify_image(photos):
-  photos = photos.reshape((-1, 256, 256, 3))
+  photos = photos.reshape((-1, 224, 224, 3))
   prediction = model.predict(photos).flatten()
   confidences = {labels[i]: float(prediction[i]) for i in range(23)}
   return confidences
@@ -53,7 +53,7 @@ gr.Interface(fn=classify_image,
              title = title,
              article = article,
              description = description, 
-             inputs=gr.inputs.Image(shape=(256, 256)),
+             inputs=gr.inputs.Image(shape=(224, 224)),
              outputs=gr.outputs.Label(num_top_classes=4),
              examples=examples).launch()