Astridkraft commited on
Commit
c69ebec
·
verified ·
1 Parent(s): 4a5728f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,6 +11,7 @@ import os
11
  import tempfile
12
  import random
13
  import re
 
14
 
15
  # === OPTIMIERTE EINSTELLUNGEN ===
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -178,8 +179,9 @@ def create_face_mask(image, bbox_coords, mode):
178
  Returns:
179
  - padded_image: skaliertes Bild mit Padding (RGB)
180
  - padded_mask: skalierte Maske mit Padding (L)
181
- - padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
182
  """
 
183
  def scale_image_and_mask_together(image, mask, target_size=512):
184
  if image is None or mask is None:
185
  raise ValueError("Bild oder Maske ist None")
@@ -237,7 +239,6 @@ def scale_image_and_mask_together(image, mask, target_size=512):
237
 
238
 
239
 
240
-
241
  # Composition Workflow nach Ausgabe ControlnetInpaint-Pipeline
242
  def enhanced_composite_with_sam(original_image, inpaint_result, original_mask,
243
  padding_info, bbox_coords, mode):
 
11
  import tempfile
12
  import random
13
  import re
14
+ from PIL import ImageFilter # Für GaussianBlur wird nur für SAM benötigt!
15
 
16
  # === OPTIMIERTE EINSTELLUNGEN ===
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
179
  Returns:
180
  - padded_image: skaliertes Bild mit Padding (RGB)
181
  - padded_mask: skalierte Maske mit Padding (L)
182
+ - padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
183
  """
184
+
185
  def scale_image_and_mask_together(image, mask, target_size=512):
186
  if image is None or mask is None:
187
  raise ValueError("Bild oder Maske ist None")
 
239
 
240
 
241
 
 
242
  # Composition Workflow nach Ausgabe ControlnetInpaint-Pipeline
243
  def enhanced_composite_with_sam(original_image, inpaint_result, original_mask,
244
  padding_info, bbox_coords, mode):