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