ekenkel commited on
Commit
8545d11
·
1 Parent(s): cce021e

updated model to include more dogs

Browse files
Files changed (2) hide show
  1. app.py +6 -1
  2. dogIdentifierModel.pkl +2 -2
app.py CHANGED
@@ -1,7 +1,12 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
 
 
3
 
4
- searchText = ('Dobermann', 'German Shepherd', 'Golden Retriever', 'Poodle')
 
 
 
5
 
6
  def classify_image(img):
7
  pred,idx,probs = learn.predict(img)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
+ import requests
4
+ import json
5
 
6
+ URL = 'https://dog.ceo/api/breeds/list/all'
7
+
8
+ result = requests.get(url = URL).json()
9
+ searchText = [val for val in result['message']]
10
 
11
  def classify_image(img):
12
  pred,idx,probs = learn.predict(img)
dogIdentifierModel.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:44b7b70b7fd45bc09cc158d189d3bc47125978b68edf5d74be99403cb5e20a23
3
- size 46980331
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6362e57d7762974fc0025b2bb372be14f33a6bbbc0d63eeb24aa1f4787666937
3
+ size 47421995