dcavadia commited on
Commit
701e439
·
1 Parent(s): 4844b80

update app script

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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')