Update app.py

app.py

@@ -39,7 +39,7 @@ def check_for_adult_content(image):
 
     return False  # No explicit content detected
 
-# Face swapping function with added smoothing
+# Face swapping function with added sharpening
 def swap_faces(destination_image, source_image):
     # Load the destination and source images from Streamlit inputs
     img = cv2.cvtColor(np.array(destination_image), cv2.COLOR_RGB2BGR)
@@ -62,19 +62,19 @@ def swap_faces(destination_image, source_image):
     except MemoryError:
         return "Memory error: Face swapping operation failed due to memory overload."
 
-    # Commented sharpening for testing purposes
-    # kernel = np.array([[0, -1, 0],
-    #                    [-1, 5, -1],
-    #                    [0, -1, 0]])
-    # sharpened_res = cv2.filter2D(res, -1, kernel)
+    # Apply sharpening for a clearer image
+    kernel = np.array([[0, -1, 0],
+                       [-1, 5, -1],
+                       [0, -1, 0]])
+    sharpened_res = cv2.filter2D(res, -1, kernel)
 
-    # Add smoothing (blurring) for a more natural look
-    smoothed_res = cv2.GaussianBlur(res, (5, 5), 0)
+    # Commented smoothing (blurring) for testing purposes
+    # smoothed_res = cv2.GaussianBlur(res, (5, 5), 0)
 
     # Convert the result to RGB for display in Streamlit
-    smoothed_res_rgb = cv2.cvtColor(smoothed_res, cv2.COLOR_BGR2RGB)
+    sharpened_res_rgb = cv2.cvtColor(sharpened_res, cv2.COLOR_BGR2RGB)
 
-    return smoothed_res_rgb
+    return sharpened_res_rgb
 
 # Streamlit app layout
 def main():