Spaces:

professorbrat
/

melanoma_classification

Build error

professorbrat commited on Sep 13, 2022

Commit

a69b26a

•

1 Parent(s): 817d742

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def classify_image(inp):
     image_array_with_batchdim = image_array[np.newaxis, :]
-    model_prediction = np.argmax(model.predict(image_array_with_batchdim), axis=-1)[0]
     return labels[model_prediction]

     image_array_with_batchdim = image_array[np.newaxis, :]
+    model_prediction = np.argmax(nomanet.predict(image_array_with_batchdim), axis=-1)[0]
     return labels[model_prediction]