Spaces:
Sleeping
Sleeping
Update difpoint/inference.py
Browse files- difpoint/inference.py +1 -2
difpoint/inference.py
CHANGED
@@ -273,7 +273,7 @@ class Inferencer(object):
|
|
273 |
# 2024.06.26
|
274 |
@torch.no_grad()
|
275 |
def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t, save_path='results'):
|
276 |
-
|
277 |
if audio_type == 'upload':
|
278 |
audio_path = upload_audio_path
|
279 |
elif audio_type == 'tts':
|
@@ -485,5 +485,4 @@ class Inferencer(object):
|
|
485 |
print('#' * 25 + f'End Inference, cost time {time.time() - st}' + '#' * 25)
|
486 |
return save_path
|
487 |
|
488 |
-
|
489 |
|
|
|
273 |
# 2024.06.26
|
274 |
@torch.no_grad()
|
275 |
def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t, save_path='results'):
|
276 |
+
print(audio_type)
|
277 |
if audio_type == 'upload':
|
278 |
audio_path = upload_audio_path
|
279 |
elif audio_type == 'tts':
|
|
|
485 |
print('#' * 25 + f'End Inference, cost time {time.time() - st}' + '#' * 25)
|
486 |
return save_path
|
487 |
|
|
|
488 |
|