Hugging Face's logo

Spaces:
ltomczak1
/
lungcancer_subclassifier
Runtime error

App Files Community
ltomczak1 commited on
Commit
c812851
•
1 Parent(s): 005a397

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ categories = ('ACA', 'Benign', 'SCC')
8
 
9
  def classify_image(img):
10
  pred,idx,probs = learn.predict(img)
11
- return dict(zip(categories, map(float,probs)))
12
-
13
  image = gr.components.Image(shape=(768, 768))
14
  label = gr.components.Label()
15
  examples = ['lungaca.jpeg', 'lungaca142.jpeg', 'lungaca199.jpeg', 'lungaca861.jpeg',
 
8
 
9
  def classify_image(img):
10
  pred,idx,probs = learn.predict(img)
11
+ return categories
12
+
13
  image = gr.components.Image(shape=(768, 768))
14
  label = gr.components.Label()
15
  examples = ['lungaca.jpeg', 'lungaca142.jpeg', 'lungaca199.jpeg', 'lungaca861.jpeg',