File size: 657 Bytes
7206ed3 6db6a87 63056c5 ea66b57 7206ed3 13575f0 6db6a87 13575f0 7206ed3 ea66b57 13575f0 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 29 30 31 32 33 |
import os
import yaml
import numpy as np
from matplotlib import cm
import gradio as gr
import deeplabcut
import dlclibrary
import transformers
from PIL import Image
import requests
#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)
|