Spaces:
Sleeping
Sleeping
loaded learner
Browse files- app.py +5 -0
- petClassifierModel.pkl +3 -0
- requirements.txt +5 -0
app.py
CHANGED
@@ -1,5 +1,10 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
|
|
|
|
|
|
|
|
|
3 |
def greet(name):
|
4 |
return "Hello " + name + "!!"
|
5 |
|
|
|
1 |
+
from fastai.vision.all import *
|
2 |
import gradio as gr
|
3 |
|
4 |
+
learn = load_learner('petClassifierModel.pkl')
|
5 |
+
|
6 |
+
categories = ("American Bulldog", "American Pit Bull Terrier", "Basset Hound", "Beagle", "Boxer", "Chihuahua", "English Cocker Spaniel", "English Setter", "German Shorthaired", "Great Pyrenees", "Havanese", "Japanese Chin", "Keeshond", "Leonberger", "Miniature Pinscher", "Newfoundland", "Pomeranian", "Pug", "Saint Bernard", "Samyoed", "Scottish Terrier", "Shiba Inu", "Staffordshire Bull Terrier")
|
7 |
+
|
8 |
def greet(name):
|
9 |
return "Hello " + name + "!!"
|
10 |
|
petClassifierModel.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:21e978392dcce268b3b4824ef58bea846141faff99f7dd9750c7c5ecb2bffb84
|
3 |
+
size 87657479
|
requirements.txt
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
fastai
|
2 |
+
torch
|
3 |
+
gradio
|
4 |
+
numpy
|
5 |
+
pandas
|