Update app.py

app.py

@@ -1,7 +1,3 @@
-# Built from https://huggingface.co/spaces/hlydecker/MegaDetector_v5 
-# Built from https://huggingface.co/spaces/sofmi/MegaDetector_DLClive/blob/main/app.py
-# Built from https://huggingface.co/spaces/Neslihan/megadetector_dlcmodels/blob/main/app.py 
-
 import os
 import yaml
 import numpy as np
@@ -9,24 +5,27 @@ from matplotlib import cm
 import gradio as gr
 import deeplabcut
 import dlclib
-
-from PIL import Image, ImageColor, ImageFont, ImageDraw 
-# check git lfs pull!!
-from DLC_models.download_utils import DownloadModel
-from dlclive import DLCLive, Processor
+import transformers
 
 
+from PIL import Image, ImageColor, ImageFont, ImageDraw 
 from viz_utils import save_results_as_json, draw_keypoints_on_image, draw_bbox_w_text, save_results_only_dlc
 from detection_utils import predict_md, crop_animal_detections, predict_dlc
 from ui_utils import gradio_inputs_for_MD_DLC, gradio_outputs_for_MD_DLC, gradio_description_and_examples
 
-# directly from huggingface
 
-import gradio as gr
+# 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=["This is a sample input."])
+
+# Print the prediction
+print(prediction)
 
-model = gr.Interface.load("mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped",
-)
 
-model.launch()