Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ from matplotlib import cm
|
|
5 |
import gradio as gr
|
6 |
import deeplabcut
|
7 |
import dlclibrary
|
|
|
8 |
import transformers
|
9 |
|
10 |
from PIL import Image
|
@@ -20,9 +21,9 @@ from dlclibrary.dlcmodelzoo.modelzoo_download import (
|
|
20 |
download_huggingface_model,
|
21 |
MODELOPTIONS,
|
22 |
)
|
23 |
-
|
24 |
from dlclive import DLCLive, Processor
|
25 |
|
|
|
26 |
# TESTING (passes) download the SuperAnimal models:
|
27 |
#model = 'superanimal_topviewmouse'
|
28 |
#train_dir = 'DLC_models/sa-tvm'
|
|
|
5 |
import gradio as gr
|
6 |
import deeplabcut
|
7 |
import dlclibrary
|
8 |
+
import dlclive
|
9 |
import transformers
|
10 |
|
11 |
from PIL import Image
|
|
|
21 |
download_huggingface_model,
|
22 |
MODELOPTIONS,
|
23 |
)
|
|
|
24 |
from dlclive import DLCLive, Processor
|
25 |
|
26 |
+
|
27 |
# TESTING (passes) download the SuperAnimal models:
|
28 |
#model = 'superanimal_topviewmouse'
|
29 |
#train_dir = 'DLC_models/sa-tvm'
|