Herrius
commited on
Commit
•
6d79ae7
1
Parent(s):
6a9b0f3
update load learner
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -2,7 +2,7 @@ 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')
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# Cell
|
| 5 |
+
learn = load_learner('.','model.pkl')
|
| 6 |
|
| 7 |
# Cell
|
| 8 |
categories = ('rocoto','pimenton')
|
requirements.txt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
transformers
|
| 2 |
torch
|
| 3 |
-
fastai
|
|
|
|
| 1 |
transformers
|
| 2 |
torch
|
| 3 |
+
fastai
|