Update app.py

app.py

@@ -1,25 +1,13 @@
 import gradio as gr
-from rembg import remove
+import cv2
 import numpy as np
 
-# Функция для удаления водяных знаков с изображения
-def remove_watermark(input_image):
-    # Сохранение массива numpy в файл
-    input_image = input_image.astype(np.uint8)
-    input_image = np.squeeze(input_image)
-    with open("input.png", "wb") as f:
-        f.write(input_image)
-
-    # Загрузка изображения и удаление фона
-    output = remove("input.png")
-
-    return output.tobytes()
-
-# Создание Gradio приложения
-iface = gr.Interface(
-    fn=remove_watermark,
-    inputs=gr.Image(type="numpy"),
-    outputs=gr.Image(),
-)
+def upscale_image(input_image):
+    # Увеличиваем размер изображения в 2 раза
+    upscale_factor = 2
+    output_image = cv2.resize(input_image, None, fx = upscale_factor, fy = upscale_factor, interpolation = cv2.INTER_CUBIC)
+    return output_image
 
+# Создаем интерфейс Gradio
+iface = gr.Interface(fn=upscale_image, inputs="image", outputs="image")
 iface.launch()