Herrius commited on
Commit
6a9b0f3
1 Parent(s): eaaaba7

code of load of model

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
 
 
4
  # Cell
5
  categories = ('rocoto','pimenton')
6
  def classify_image(img):
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ # Cell
5
+ learn = load_learner('model.pkl')
6
+
7
  # Cell
8
  categories = ('rocoto','pimenton')
9
  def classify_image(img):