abreza commited on
Commit
e1fd5b9
1 Parent(s): 70ec0b5

add some logs

Browse files
Files changed (1) hide show
  1. app.py +1 -11
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