tomaseo2022's picture
Rename app.py to app2.py
b91a567 verified
raw
history blame
1.63 kB
import os
os.system("pip install --upgrade httpx")
os.system("pip install --upgrade gradio")
os.system("pip install opencv-python")
os.system("pip install torch")
os.system("pip install --upgrade pillow")
os.system("pip install numpy")
import gradio as gr
from PIL import Image
import torch
import numpy as np
# Instalar paquetes necesarios
os.system("pip install --upgrade httpx gradio opencv-python torch pillow numpy")
# Descargar modelo pre-entrenado
os.system('wget https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth -P experiments/pretrained_models')
def inference(img):
try:
# Verificar si la entrada es una imagen v谩lida
if not isinstance(img, Image.Image):
return "Error: La entrada no es una imagen v谩lida. Por favor, proporcione una imagen v谩lida."
# Redimensionar la imagen
basewidth = 256
wpercent = basewidth / float(img.width)
hsize = int(float(img.height) * float(wpercent))
img_resized = img.resize((basewidth, hsize))
# L贸gica para mejorar la resoluci贸n de la imagen
# (Puedes agregar tu l贸gica aqu铆)
# Devolver la imagen procesada
return img_resized
except Exception as e:
return "Error: Ha ocurrido un error durante el procesamiento de la imagen. Por favor, int茅ntelo de nuevo con una imagen diferente."
iface = gr.Interface(
fn=inference,
inputs="image",
outputs="image",
title="Mejorar Resoluci贸n de Imagen",
description="Sube una imagen y mejora su resoluci贸n."
)
iface.launch(debug=True)