tomaseo2022 commited on
Commit
dcc3c3a
Β·
1 Parent(s): e8d7c25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -11
app.py CHANGED
@@ -7,19 +7,23 @@ import cv2
7
  import numpy as np
8
 
9
  def image_to_silhouette(image):
10
- # Convertir imagen a escala de grises
11
- gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
 
12
 
13
- # Aplicar bordes
14
- edges = cv2.Canny(gray, 50, 150)
15
 
16
- # Crear mΓ‘scara con forma de silueta negra
17
- silhouette = np.zeros_like(image)
18
- silhouette[:,:,0] = edges
19
- silhouette[:,:,1] = edges
20
- silhouette[:,:,2] = edges
21
 
22
- return silhouette
 
 
 
23
 
24
  # Crear interfaz de usuario con Gradio
25
  gr.Interface(
@@ -27,4 +31,4 @@ gr.Interface(
27
  gr.inputs.Image(type="filepath", label="Input"),
28
  gr.outputs.Image(type="pil", label="Output"),
29
  css="Footer {visibility: hidden}"
30
- ).launch()
 
7
  import numpy as np
8
 
9
  def image_to_silhouette(image):
10
+ try:
11
+ # Convertir imagen a escala de grises
12
+ gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
13
 
14
+ # Aplicar bordes
15
+ edges = cv2.Canny(gray, 50, 150)
16
 
17
+ # Crear mΓ‘scara con forma de silueta negra
18
+ silhouette = np.zeros_like(image)
19
+ silhouette[:,:,0] = edges
20
+ silhouette[:,:,1] = edges
21
+ silhouette[:,:,2] = edges
22
 
23
+ return silhouette
24
+ except Exception as e:
25
+ print("An error occurred:", e)
26
+ return None
27
 
28
  # Crear interfaz de usuario con Gradio
29
  gr.Interface(
 
31
  gr.inputs.Image(type="filepath", label="Input"),
32
  gr.outputs.Image(type="pil", label="Output"),
33
  css="Footer {visibility: hidden}"
34
+ ).launch()