Hugging Face's logo

Spaces:
archietram
/
Diabetic_Retinopathy_Predictor
Runtime error

App Files Community
archietram commited on
Commit
07dd9a9
•
1 Parent(s): 3340333

Update app.py

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