Update app.py

app.py

@@ -17,7 +17,7 @@ def load_model_and_predict(image, model_name):
         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)
+        (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)
         (out, _) = infer_image(net, image, transform)