DaniloTertu commited on
Commit
8d4e4e1
·
1 Parent(s): a2be937

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import fastai
3
 
4
  #__all__ = ['NumberClass', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
5
 
@@ -7,7 +7,7 @@ def get_y(r):
7
  number = ''.join(filter(str.isdigit, str(r)))
8
  return train_y[int(number)]
9
 
10
- learn = load_learner('CollaborativeFiltering.pkl')
11
 
12
  label = gr.outputs.Label()
13
 
 
1
  import gradio as gr
2
+ import joblib
3
 
4
  #__all__ = ['NumberClass', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
5
 
 
7
  number = ''.join(filter(str.isdigit, str(r)))
8
  return train_y[int(number)]
9
 
10
+ learn = joblib.load('CollaborativeFiltering.pkl')
11
 
12
  label = gr.outputs.Label()
13
 
requirements.txt CHANGED
@@ -1 +1 @@
1
- fastai
 
1
+ joblib