archietram commited on
Commit
8c3cefe
1 Parent(s): f8acfd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ learn = load_learner("export.pkl")
6
 
7
  categories = ("Diabetic Retinopathy", "No Diabetic Retinopathy")
8
 
 
 
9
  def classify_image(img):
10
  pred,idx,probs = learn.predict(img)
11
  return dict(zip(categories, map(float,probs)))
 
6
 
7
  categories = ("Diabetic Retinopathy", "No Diabetic Retinopathy")
8
 
9
+ def get_disease(): return _
10
+
11
  def classify_image(img):
12
  pred,idx,probs = learn.predict(img)
13
  return dict(zip(categories, map(float,probs)))