Spaces:
Running
Running
dcavadia commited on
Commit ·
701e439
1
Parent(s): 4844b80
update app script
Browse files
app.py
CHANGED
|
@@ -2,9 +2,12 @@ import json
|
|
| 2 |
from keras.models import Model, load_model
|
| 3 |
import gradio as gr
|
| 4 |
import cv2
|
|
|
|
| 5 |
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
-
model = load_model('model.h5', compile=True)
|
| 8 |
|
| 9 |
# Opening JSON file
|
| 10 |
f = open('dat.json')
|
|
|
|
| 2 |
from keras.models import Model, load_model
|
| 3 |
import gradio as gr
|
| 4 |
import cv2
|
| 5 |
+
from fastai.learner import load_learner
|
| 6 |
|
| 7 |
+
# Load the exported learner
|
| 8 |
+
learn = load_learner('nfnet_model_20epoch_adam_0.962.pkl')
|
| 9 |
+
model = learn.model
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
# Opening JSON file
|
| 13 |
f = open('dat.json')
|