Spaces:

asaripa3
/

worst_race_detector

Sleeping

asaripa3 commited on Jun 19, 2024

Commit

5d251cd

verified ·

1 Parent(s): 01c19fc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ learn = load_learner("race_model.pkl")
 # In[72]:
 #|export
-categories = ("Indian", "American", "Korean", "African", "Chinese")
 def classify_image(img):

 # In[72]:
 #|export
+categories = learn.dls.vocab
 def classify_image(img):