Frantz103 commited on
Commit
1d11d2b
1 Parent(s): 9ecc50a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,6 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
- learn = load_learner('dog_cat_multi.pkl')
5
-
6
  cat_breeds = [
7
  'Abyssinian', 'Bengal', 'Birman', 'Bombay', 'British_Shorthair',
8
  'Egyptian_Mau', 'Main_Coon', 'Persian', 'Ragdoll', 'Russian_Blue',
@@ -21,6 +19,8 @@ def get_labels(path):
21
  print(f"Could not match pattern in {path.name}")
22
  return []
23
 
 
 
24
  def predict(image):
25
  # Transform the image
26
  pil_image = PILImage.create(image)
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
 
 
4
  cat_breeds = [
5
  'Abyssinian', 'Bengal', 'Birman', 'Bombay', 'British_Shorthair',
6
  'Egyptian_Mau', 'Main_Coon', 'Persian', 'Ragdoll', 'Russian_Blue',
 
19
  print(f"Could not match pattern in {path.name}")
20
  return []
21
 
22
+ learn = load_learner('dog_cat_multi.pkl')
23
+
24
  def predict(image):
25
  # Transform the image
26
  pil_image = PILImage.create(image)