uhlarlar commited on
Commit
dcb2810
·
1 Parent(s): dfbac7d

loaded learner

Browse files
Files changed (3) hide show
  1. app.py +5 -0
  2. petClassifierModel.pkl +3 -0
  3. 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