Update app.py

app.py

@@ -14,6 +14,9 @@ def get_birder_classification_models():
 
 def load_model_and_predict(image, model_name):
     try:
+        if len(birder.list_pretrained_models(model_name)) == 0:
+            model_name = birder.list_pretrained_models(model_name + "*")[0]
+
         (net, class_to_idx, signature, rgb_stats) = birder.load_pretrained_model(model_name, inference=True)
         size = birder.get_size_from_signature(signature)
         transform = birder.classification_transform(size, rgb_stats)