added dlc

app.py

@@ -31,6 +31,13 @@ def yolo(im): #size=640):
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])
 
+# %% dlc-live function
+def dlc(im):
+    from dlclive import DLCLive, Processor
+    dlc_proc = Processor()
+    dlc_live = DLCLive(https://huggingface.co/spaces/vchiang001/MegaDetector_DLC_pose/blob/main/quadraped_snapshot-1000.pb, processor=dlc_proc)
+    dlc_live.init_inference(im)
+    dlc_live.get_pose(im)
 
 # %% 
 #inputs = [image, chosen_model, size]
@@ -46,4 +53,4 @@ article = "<p style='text-align: center'>Detect and identify animals, people and
 
 # %% 
 #running the actual 
-gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, theme="huggingface").launch(enable_queue=True)
+gr.Interface(dlc, inputs, outputs, title=title, description=description, article=article, theme="huggingface").launch(enable_queue=True)