Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|