fdsa

app.py

@@ -6,7 +6,13 @@ import timm
 
 # quick function to strip the leading ###. off the parent label name
 def strip_parent_num(o):
-  return parent_label(o).split('.')[1]
+  #return parent_label(o).split('.')[1]
+  try:
+    r = parent_label(o).split('.')[1]
+  except IndexError:
+    r = parent_label(o)
+
+  return r
 
 # load the model
 learn = load_learner('birds-convnext_small_in22k.pkl')
@@ -22,8 +28,7 @@ def predict(img):
 # Gradio customizations
 title =  'Bird Identifier'
 description = 'This model will predict the type of bird from an image.\n\nThe convnext_tiny_in22k model was refined using the CUB_200_2011 bird dataset.'
-#examples = ['cardinal1.jpg', 'crow.jpg']
-examples = ['siamese.jpg']
+examples = ['cardinal1.jpg', 'crow.jpg']
 interpretation = 'default'
 enable_queue = True
 article = '''<p style='text-align: center'><a href='https://tmabraham.github.io/blog/gradio_hf_spaces_tutorial' target='_blank'>Learning from -> Gradio + HuggingFace Spaces: A Tutorial</a></p>'''