app.py

import os
import yaml
import numpy as np
from matplotlib import cm
import gradio as gr
import deeplabcut
import dlclib
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


# 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)