Update txt2vid.py

pull/191/head
alexbofa 2023-08-31 18:42:48 +03:00 committed by GitHub
parent 6a04d70241
commit de1ff473f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -6,7 +6,7 @@ import numpy as np
from PIL import Image from PIL import Image
import modules.paths as ph import modules.paths as ph
from modules.shared import devices from modules import devices
from scripts.core import utils, flow_utils from scripts.core import utils, flow_utils
from FloweR.model import FloweR from FloweR.model import FloweR
@ -237,4 +237,4 @@ def start_process(*args):
# print('TOTAL TIME:', int(time.time() - processing_start_time)) # print('TOTAL TIME:', int(time.time() - processing_start_time))
yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False) yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False)