Spaces:

professorbrat
/

melanoma_classification

Build error

professorbrat commited on Sep 13, 2022

Commit

2b5dc41

•

1 Parent(s): c3e7cdd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def classify_image(inp):
     predictions = nomanet.predict(img_array)[0]
-    return labels[model_prediction]
 gr.Interface(fn=classify_image,
              inputs=gr.inputs.Image(type="filepath"),

     predictions = nomanet.predict(img_array)[0]
+    return labels[predictions]
 gr.Interface(fn=classify_image,
              inputs=gr.inputs.Image(type="filepath"),