Spaces:
Running
on
Zero
Running
on
Zero
add some logs
Browse files
app.py
CHANGED
@@ -184,19 +184,9 @@ def test(source_image, driven_audio, preprocess='crop',
|
|
184 |
return_path = animate_from_coeff.generate(
|
185 |
data, save_dir, pic_path, crop_info, enhancer='gfpgan' if use_enhancer else None, preprocess=preprocess, img_size=size)
|
186 |
video_name = data['video_name']
|
|
|
187 |
print(f'The generated video is named {video_name} in {save_dir}')
|
188 |
|
189 |
-
del preprocess_model
|
190 |
-
del audio_to_coeff
|
191 |
-
del animate_from_coeff
|
192 |
-
|
193 |
-
if torch.cuda.is_available():
|
194 |
-
torch.cuda.empty_cache()
|
195 |
-
torch.cuda.synchronize()
|
196 |
-
|
197 |
-
import gc
|
198 |
-
gc.collect()
|
199 |
-
|
200 |
return return_path
|
201 |
|
202 |
|
|
|
184 |
return_path = animate_from_coeff.generate(
|
185 |
data, save_dir, pic_path, crop_info, enhancer='gfpgan' if use_enhancer else None, preprocess=preprocess, img_size=size)
|
186 |
video_name = data['video_name']
|
187 |
+
print('!!!!!')
|
188 |
print(f'The generated video is named {video_name} in {save_dir}')
|
189 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
190 |
return return_path
|
191 |
|
192 |
|