File size: 501 Bytes
7206ed3 6db6a87 63056c5 ea66b57 7206ed3 6db6a87 7206ed3 ea66b57 7206ed3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
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)
|