File size: 845 Bytes
7206ed3 6db6a87 ea66b57 7206ed3 6db6a87 ea66b57 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 29 30 31 32 |
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)
|