github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-20 13:33:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix FaceSwapper.py

Browse source
This commit is contained in:
iperov 2021-08-06 20:41:27 +04:00
commit 9342ada240
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
apps/DeepFaceLive/backend/FaceSwapper.py
View file

@ -156,9 +156,6 @@ class FaceSwapperWorker(BackendWorker):
def on_tick(self): def on_tick(self):
state, cs = self.get_state(), self.get_control_sheet() state, cs = self.get_state(), self.get_control_sheet()
@ -166,12 +163,15 @@ class FaceSwapperWorker(BackendWorker):
events = self.dfm_model_initializer.process_events() events = self.dfm_model_initializer.process_events()
if events.prev_status_downloading: if events.prev_status_downloading:
self.set_busy(True)
cs.model_dl_progress.disable() cs.model_dl_progress.disable()
if events.new_status_downloading: if events.new_status_downloading:
self.set_busy(False)
cs.model_dl_progress.enable() cs.model_dl_progress.enable()
cs.model_dl_progress.set_config( lib_csw.Progress.Config(title='@FaceSwapper.downloading_model') ) cs.model_dl_progress.set_config( lib_csw.Progress.Config(title='@FaceSwapper.downloading_model') )
cs.model_dl_progress.set_progress(0) cs.model_dl_progress.set_progress(0)
elif events.new_status_initialized: elif events.new_status_initialized:
self.dfm_model = events.dfm_model self.dfm_model = events.dfm_model
self.dfm_model_initializer = None self.dfm_model_initializer = None