Update app.py

app.py

@@ -5,7 +5,6 @@ import shutil
 import requests
 import numpy as np
 from PIL import Image, ImageOps
-import cv2
 import math
 import matplotlib.pyplot as plt
 import pickle
@@ -44,20 +43,7 @@ def gen_sources(deepfake_img):
     #----------------DeepFake Face Segmentation-----------------
     ##----------------------Initialize:Face Segmentation----------------------------------
     segmenter = FaceSegmenter(threshold=0.5)
-    #Convert Pillow Image to OpenCV Image (BGR format)
-    dfcv2_image = np.array(deepfake_img)  # Convert PIL to numpy array (RGB)
-    dfcv2_image = cv2.cvtColor(dfcv2_image, cv2.COLOR_RGB2BGR)  # Convert RGB to BGR
-    deepfake_seg = segmenter.segment_face(dfcv2_image)
-    # Save PIL Image to temporary file
-    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as temp_file:
-        temp_path = temp_file.name
-        deepfake_img.save(temp_path, format="JPEG")
-    
-    # Segment face using the temporary file path
-    try:
-        deepfake_seg = segmenter.segment_face(temp_path)
-    finally:
-        os.remove(temp_path)  # Clean up temporary file
+    deepfake_seg = segmenter.segment_face(deepfake_img)
         
     config_path = "./models/config.yaml"
     #------------Initialize:Decoder-F------------------------