dibend commited on
Commit
dc80f01
·
verified ·
1 Parent(s): ce8d5bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -14
app.py CHANGED
@@ -216,18 +216,6 @@ def _create_sunglasses_mask(style="aviator"):
216
  # Bridge
217
  draw.rectangle((130, 60, 170, 75), fill='white')
218
 
219
- elif style == "heart_shaped":
220
- # Left Heart
221
- draw.polygon([
222
- (75, 40), (125, 60), (75, 120), (25, 60)
223
- ], fill='deeppink', outline='hotpink', width=4)
224
- # Right Heart
225
- draw.polygon([
226
- (225, 40), (275, 60), (225, 120), (175, 60)
227
- ], fill='deeppink', outline='hotpink', width=4)
228
- # Bridge
229
- draw.line((125, 70, 175, 70), fill='hotpink', width=6)
230
-
231
  return sunglasses
232
 
233
  def apply_sunglasses(img_np, style="aviator"):
@@ -271,7 +259,6 @@ def process_image(image, filter_name, splash_color):
271
  "Color Splash": lambda img: apply_color_splash(img, splash_color),
272
  "Aviator Sunglasses": lambda img: apply_sunglasses(img, style="aviator"),
273
  "Retro Square Sunglasses": lambda img: apply_sunglasses(img, style="retro_square"),
274
- "Heart-Shaped Sunglasses": lambda img: apply_sunglasses(img, style="heart_shaped"),
275
  "None": lambda img: img
276
  }
277
 
@@ -292,7 +279,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
292
  filters_standard = ["Grayscale", "Sepia", "Invert", "Posterize", "Solarize", "Vignette", "Contour", "Sharpen"]
293
  filters_artistic = ["Cartoon", "Sketch", "Pixelate"]
294
  filters_advanced = ["N64-ify (Triangles)", "HDR Effect", "Color Splash", "Sunburst Glow", "Dreamy Glow"]
295
- filters_face = ["Aviator Sunglasses", "Retro Square Sunglasses", "Heart-Shaped Sunglasses"]
296
 
297
  all_filters = ["None"] + filters_standard + filters_artistic + filters_advanced + filters_face
298
 
 
216
  # Bridge
217
  draw.rectangle((130, 60, 170, 75), fill='white')
218
 
 
 
 
 
 
 
 
 
 
 
 
 
219
  return sunglasses
220
 
221
  def apply_sunglasses(img_np, style="aviator"):
 
259
  "Color Splash": lambda img: apply_color_splash(img, splash_color),
260
  "Aviator Sunglasses": lambda img: apply_sunglasses(img, style="aviator"),
261
  "Retro Square Sunglasses": lambda img: apply_sunglasses(img, style="retro_square"),
 
262
  "None": lambda img: img
263
  }
264
 
 
279
  filters_standard = ["Grayscale", "Sepia", "Invert", "Posterize", "Solarize", "Vignette", "Contour", "Sharpen"]
280
  filters_artistic = ["Cartoon", "Sketch", "Pixelate"]
281
  filters_advanced = ["N64-ify (Triangles)", "HDR Effect", "Color Splash", "Sunburst Glow", "Dreamy Glow"]
282
+ filters_face = ["Aviator Sunglasses", "Retro Square Sunglasses"]
283
 
284
  all_filters = ["None"] + filters_standard + filters_artistic + filters_advanced + filters_face
285