moshe742 commited on
Commit
3cf0156
·
1 Parent(s): 32c4a8c

add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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 = learn.predict(image)
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