Hugging Face's logo

Spaces:
Frantz103
/
dog_cat_multi_classifier
Sleeping

App Files Community
Frantz103 commited on
Commit
d3e9dc6
•
1 Parent(s): f3ae7da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import gradio as gr
2
  from fastai.vision.all import PILImage
3
 
 
4
 
 
5
 
6
  def predict(image):
7
  # Transform the image
 
1
  import gradio as gr
2
  from fastai.vision.all import PILImage
3
 
4
+ from fastai.learner import load_learner
5
 
6
+ learn = load_learner(dog_cat_multi.pkl')
7
 
8
  def predict(image):
9
  # Transform the image