import gradio as gr
import torch
from PIL import Image

# Função simples para rodar o modelo
def run_model(image):
    # Aqui você pode integrar a lógica do seu pipeline de modelo (exemplo de uso)
    # Como exemplo, vamos apenas redimensionar a imagem para "simular" o processo.
    image = image.resize((512, 512))
    return image

# Criando o interface Gradio
iface = gr.Interface(fn=run_model, inputs=gr.Image(), outputs=gr.Image())

# Rodando a interface
iface.launch()