DaniloTertu commited on
Commit
c83ef54
·
1 Parent(s): 2f0c909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import joblib
3
  from fastai.collab import *
4
  from fastai.tabular.all import *
5
 
@@ -24,7 +24,9 @@ def get_y(r):
24
  number = ''.join(filter(str.isdigit, str(r)))
25
  return train_y[int(number)]
26
 
27
- learn = joblib.dump('collaborativefiltering.pkl')
 
 
28
 
29
  label = gr.outputs.Label()
30
 
 
1
  import gradio as gr
2
+ import pickle
3
  from fastai.collab import *
4
  from fastai.tabular.all import *
5
 
 
24
  number = ''.join(filter(str.isdigit, str(r)))
25
  return train_y[int(number)]
26
 
27
+
28
+ with open('collaborativefiltering.pkl', 'rb') as f:
29
+ learn = pickle.load(f)
30
 
31
  label = gr.outputs.Label()
32