Spaces:

BimanL
/

huawei_seeds_for_tommrrow

Sleeping

BimanL commited on Nov 16, 2023

Commit

4943c07

•

1 Parent(s): 1524fdc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,10 @@
 import gradio as gr
 from fastai.vision.all import *
 import skimage
 # Load your trained model
-learn = load_learner('export.pkl')
 # Define the labels (in this case, just 'cat' and 'dog')
 labels = learn.dls.vocab

 import gradio as gr
+from fastai.vision.widgets import *
 from fastai.vision.all import *
 import skimage
 # Load your trained model
+learn = load_learner('./export.pkl')
 # Define the labels (in this case, just 'cat' and 'dog')
 labels = learn.dls.vocab