Update app.py

app.py

@@ -40,6 +40,8 @@ transform = T.Compose([
 
 def gen_sources(deepfake_img):
     #----------------DeepFake Face Segmentation-----------------
+    ##----------------------Initialize:Face Segmentation----------------------------------
+    segmenter = FaceSegmenter(threshold=0.5)
     deepfake_seg = segmenter.segment_face(deepfake_img)
     config_path = "./models/config.yaml"
     #------------Initialize:Decoder-F------------------------
@@ -64,8 +66,6 @@ def gen_sources(deepfake_img):
     model_z2.eval()
     ##--------------------Initialize:Evaluation---------------------------------------
     criterion = DF()
-    ##----------------------Initialize:Face Segmentation----------------------------------
-    segmenter = FaceSegmenter(threshold=0.5)
     
     ##----------------------Operation-------------------------------------------------
     with torch.no_grad():