Spaces:

MasleK
/

Snails_Snakes_Slugs

Sleeping

MasleK commited on Apr 26, 2023

Commit

8b64710

1 Parent(s): ca567b7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,12 +6,12 @@ from huggingface_hub import from_pretrained_fastai
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
 repo_id = "MasleK/snails_snakes_slugs"
-learner = from_pretrained_fastai(repo_id)
-categories = learner.learn.dls.vocab
 def predict(image):
-    label, index, probs = learner.predict(image)
     return dict(zip(categories, map(float,probs)))

 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
 repo_id = "MasleK/snails_snakes_slugs"
+learn = from_pretrained_fastai(repo_id)
+categories = learn.dls.vocab
 def predict(image):
+    label, index, probs = learn.predict(image)
     return dict(zip(categories, map(float,probs)))