Update app.py

app.py

@@ -6,15 +6,14 @@ import gradio as gr
 labels = {0: "melanoma", 1: "nevus", 2: "seborrheic_keratosis"}
 
 def classify_image(inp):
-    img = tf.keras.utils.load_img(
-    inp, target_size=(256,256)
-    )
-    img_array = tf.keras.utils.img_to_array(img)
-    img_array = tf.expand_dims(img_array, 0) 
+    image_array = tf.keras.preprocessing.image.img_to_array(
+                      tf.keras.preprocessing.image.load_img(inp, target_size=(256,256)))
 
-    predictions = nomanet.predict(img_array)[0]
+    image_array_with_batchdim = image_array[np.newaxis, :]
     
-    return labels[predictions]
+    model_prediction = np.argmax(model.predict(image_array_with_batchdim), axis=-1)[0]
+
+    return labels[model_prediction]
     
 gr.Interface(fn=classify_image, 
              inputs=gr.inputs.Image(type="filepath"),