Silueta / app.py
tomaseo2022's picture
Update app.py
ffdb3e9
raw
history blame
1.17 kB
import gradio as gr
import os
os.system('/usr/local/bin/python -m pip install --upgrade pip')
os.system("pip install opencv-python")
os.system("pip install numpy")
import cv2
import numpy as np
def image_to_silhouette(image):
# Convertir imagen a escala de grises
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# Aplicar bordes
edges = cv2.Canny(gray, 50, 150)
# Crear m谩scara con forma de silueta negra
silhouette = np.zeros_like(image)
silhouette[:,:,0] = edges
silhouette[:,:,1] = edges
silhouette[:,:,2] = edges
return silhouette
# Crear interfaz de usuario con Gradio
interface = gr.Interface(fn=image_to_silhouette, inputs="image", outputs="image")
# A帽adir cajas de texto y bot贸n
input_box = interface.input_components[0]
output_box = interface.output_components[0]
convert_button = gr.Button("Convertir")
# Funci贸n para manejar el evento de clic del bot贸n
def on_button_clicked(sender):
# Obtener imagen de la caja de entrada
image = input_box.value
# Convertir imagen a silueta y mostrar en la caja de salida
silhouette = image_to_silhouette(image)
output_box.value = silhouette
# Muestra la interfaz
interface.launch