Herrius
commited on
Commit
•
6a9b0f3
1
Parent(s):
eaaaba7
code of load of model
Browse files
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):
|