Update app.py

app.py

@@ -2,29 +2,33 @@ import gradio as gr
 from PIL import Image
 import numpy as np
 from realesrgan import RealESRGANer
+import os
+
+# Проверка наличия модели
+def check_model_path(model_path):
+    if not os.path.exists(model_path):
+        raise FileNotFoundError(f"Модель не найдена по пути: {model_path}. Скачайте модель и поместите её в папку 'weights'.")
 
 # Инициализация модели Real-ESRGAN
 def load_model():
+    model_path = "weights/RealESRGAN_x4plus.pth"  # Путь к модели
+    check_model_path(model_path)  # Проверка наличия модели
+    
     model = RealESRGANer(
-        scale=4,  # Масштаб увеличения (4x)
-        model_path="weights/RealESRGAN_x4plus.pth",  # Путь к модели
+        scale=4,
+        model_path=model_path,
         model=None,
-        tile=0,  # Размер тайла (0 для отключения)
+        tile=0,
         tile_pad=10,
         pre_pad=0,
-        half=False  # Использование полной точности (не half precision)
+        half=False
     )
     return model
 
 # Функция для увеличения разрешения изображения
 def upscale_image(image):
-    # Преобразование изображения в numpy array
     img = np.array(image)
-    
-    # Увеличение разрешения с помощью модели
     output, _ = model.enhance(img, outscale=4)
-    
-    # Преобразование результата обратно в изображение
     return Image.fromarray(output)
 
 # Загрузка модели