Spaces:

ltomczak1
/

lungcancer_subclassifier

Runtime error

ltomczak1 commited on Nov 1, 2022

Commit

7f09935

•

1 Parent(s): c812851

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ categories = ('ACA', 'Benign', 'SCC')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
-    return categories
 image = gr.components.Image(shape=(768, 768))
 label = gr.components.Label()

 def classify_image(img):
     pred,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float,probs)))
 image = gr.components.Image(shape=(768, 768))
 label = gr.components.Label()