abhishek HF staff commited on
Commit
f054436
1 Parent(s): 9915c5d
Files changed (3) hide show
  1. app.py +0 -7
  2. packages.txt +1 -0
  3. requirements.txt +4 -0
app.py CHANGED
@@ -1,12 +1,7 @@
1
  import gradio as gr
2
  import subprocess
3
 
4
- import matplotlib
5
-
6
- matplotlib.use("Agg")
7
- import os, sys
8
  import yaml
9
- from argparse import ArgumentParser
10
  from tqdm import tqdm
11
 
12
  import imageio
@@ -19,8 +14,6 @@ from sync_batchnorm import DataParallelWithCallback
19
  from modules.generator import OcclusionAwareGenerator
20
  from modules.keypoint_detector import KPDetector
21
  from animate import normalize_kp
22
- from scipy.spatial import ConvexHull
23
- from functools import partial
24
 
25
 
26
  def load_checkpoints(config_path, checkpoint_path, cpu=False):
1
  import gradio as gr
2
  import subprocess
3
 
 
 
 
 
4
  import yaml
 
5
  from tqdm import tqdm
6
 
7
  import imageio
14
  from modules.generator import OcclusionAwareGenerator
15
  from modules.keypoint_detector import KPDetector
16
  from animate import normalize_kp
 
 
17
 
18
 
19
  def load_checkpoints(config_path, checkpoint_path, cpu=False):
packages.txt ADDED
@@ -0,0 +1 @@
 
1
+ ffmpeg
requirements.txt CHANGED
@@ -1 +1,5 @@
 
1
  imageio-ffmpeg
 
 
 
1
+ imageio
2
  imageio-ffmpeg
3
+ tqdm
4
+ scikit-image
5
+ pyyaml