Spaces:
Sleeping
Sleeping
add requirements.txt
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
@@ -9,7 +9,7 @@ learner = from_pretrained_fastai(repo_id)
|
|
9 |
|
10 |
|
11 |
def classify_image(image):
|
12 |
-
pred, idx, probs =
|
13 |
return dict<(zip(categories, map(float, probs)))
|
14 |
|
15 |
|
|
|
9 |
|
10 |
|
11 |
def classify_image(image):
|
12 |
+
pred, idx, probs = learner.predict(image)
|
13 |
return dict<(zip(categories, map(float, probs)))
|
14 |
|
15 |
|
requirements.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
fastai==2.7.15
|