ekenkel commited on
Commit
19e03dc
·
1 Parent(s): e0e9bd9

attempt fix for build error

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. modelCreation.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def classify_image(img):
21
 
22
  learn = load_learner('dogIdentifierModel.pkl')
23
 
24
- image = gr.components.Image(shape=(192, 192))
25
  label = gr.components.Label()
26
  examples = ['golden-retriever.jpg', 'german-shepherd.jpg', 'doberman.jpg', 'husky.jpg']
27
 
 
21
 
22
  learn = load_learner('dogIdentifierModel.pkl')
23
 
24
+ image = gr.components.Image(shape=(224, 224))
25
  label = gr.components.Label()
26
  examples = ['golden-retriever.jpg', 'german-shepherd.jpg', 'doberman.jpg', 'husky.jpg']
27
 
modelCreation.py CHANGED
@@ -54,7 +54,7 @@ for o in searchText:
54
  try:
55
  dest = (path / o)
56
  dest.mkdir(exist_ok=True, parents=True)
57
- results = search_images_bing(key, f'{o} dog photo')
58
  download_images(dest, urls=results.attrgot('contentUrl'))
59
  resize_images(path/o, max_size=460, dest=path/o)
60
  except shutil.SameFileError:
 
54
  try:
55
  dest = (path / o)
56
  dest.mkdir(exist_ok=True, parents=True)
57
+ results = search_images_bing(key, f'{o} dog photos')
58
  download_images(dest, urls=results.attrgot('contentUrl'))
59
  resize_images(path/o, max_size=460, dest=path/o)
60
  except shutil.SameFileError: