PranomVignesh commited on
Commit
aaa6c62
·
1 Parent(s): b57d730

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -14,7 +14,11 @@ def predict(image):
14
  predictions = imageClassifier(image)
15
  print(predictions)
16
 
17
- return predictions
 
 
 
 
18
 
19
 
20
  title = "Detecting Tumors in MRI Images"
 
14
  predictions = imageClassifier(image)
15
  print(predictions)
16
 
17
+ output = {}
18
+ for item in predictions:
19
+ output[item['label']] = item['score']
20
+
21
+ return output
22
 
23
 
24
  title = "Detecting Tumors in MRI Images"