Update app.py

app.py

@@ -1,30 +1,31 @@
 import gradio as gr
 from PIL import Image
-import torch
-from torchvision import transforms
+import numpy as np
+from realesrgan import RealESRGANer
 
-# Загрузка предобученной модели (например, ESRGAN)
-# В данном примере используется заглушка, замените на реальную модель
+# Инициализация модели Real-ESRGAN
 def load_model():
-    # Пример загрузки модели (замените на вашу модель)
-    model = torch.hub.load('pytorch/vision:v0.10.0', 'esrgan', pretrained=True)
-    model.eval()
+    model = RealESRGANer(
+        scale=4,  # Масштаб увеличения (4x)
+        model_path="weights/RealESRGAN_x4plus.pth",  # Путь к модели
+        model=None,
+        tile=0,  # Размер тайла (0 для отключения)
+        tile_pad=10,
+        pre_pad=0,
+        half=False  # Использование полной точности (не half precision)
+    )
     return model
 
 # Функция для увеличения разрешения изображения
 def upscale_image(image):
-    # Преобразование изображения в тензор
-    input_image = transforms.ToTensor()(image).unsqueeze(0)
+    # Преобразование изображения в numpy array
+    img = np.array(image)
     
     # Увеличение разрешения с помощью модели
-    with torch.no_grad():
-        output_image = model(input_image)
+    output, _ = model.enhance(img, outscale=4)
     
-    # Преобразование тензора обратно в изображение
-    output_image = output_image.squeeze(0).clamp(0, 1)
-    output_image = transforms.ToPILImage()(output_image)
-    
-    return output_image
+    # Преобразование результата обратно в изображение
+    return Image.fromarray(output)
 
 # Загрузка модели
 model = load_model()
@@ -35,7 +36,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil", label="Входное изображение"),
     outputs=gr.Image(type="pil", label="Увеличенное изображение"),
     title="AI Image Upscaler",
-    description="Загрузите изображение, чтобы увеличить его разрешение с помощью AI."
+    description="Загрузите изображение, чтобы увеличить его разрешение с помощью Real-ESRGAN."
 )
 
 # Запуск интерфейса