Update app.py

app.py

@@ -10,22 +10,20 @@ import transformers
 from PIL import Image
 import requests
 
+from deeplabcut.utils import auxiliaryfunctions
+from dlclibrary.dlcmodelzoo.modelzoo_download import (
+    download_huggingface_model,
+    MODELOPTIONS,
+)
+
+model = 'superanimal_topviewmouse'
+download_huggingface_model(model, train_dir)
+
 #demo data cooco cat:
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
 
-# Load the model
-model = transformers.AutoModel.from_pretrained("mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped")
-
-# Create an instance of the model
-inference_api = transformers.InferenceAPI(model)
-
-# Make a prediction with the model
-prediction = inference_api.predict(inputs=image)
-
-# Print the prediction
-print(prediction)