Update app.py

app.py

@@ -1,24 +1,29 @@
-from rembg import remove
 import gradio as gr
 from PIL import Image
-import io
+import torch
+from deoldify import device
+from deoldify.device_id import DeviceId
+from deoldify.visualize import get_image_colorizer
 
-# Arka planı temizleme fonksiyonu
-def remove_background(image):
-    output = remove(image)
-    return output
+device.set(device=DeviceId.GPU0 if torch.cuda.is_available() else DeviceId.CPU)
+colorizer = get_image_colorizer(artistic=True)
+
+def colorize_image(image, render_factor):
+    result = colorizer.get_transformed_image(image, render_factor=render_factor)
+    return result
 
-# Gradio arayüzü
 interface = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(type="pil", label="Görsel Yükle"),
-    outputs=gr.Image(type="pil", label="Arka Planı Temizlenmiş Görsel"),
-    title="Arka Plan Temizleme Uygulaması",
-    description="Bir görsel yükleyin ve arka planını otomatik olarak temizleyin!",
+    fn=colorize_image,
+    inputs=[
+        gr.Image(type="pil", label="Siyah Beyaz Görsel Yükle"),
+        gr.Slider(minimum=10, maximum=50, default=35, step=1, label="Renklendirme Yoğunluğu")
+    ],
+    outputs=gr.Image(type="pil", label="Renklendirilmiş Görsel"),
+    title="Siyah Beyaz Görsel Renklendirme",
+    description="Siyah beyaz bir görsel yükleyin ve renklendirme yoğunluğunu ayarlayarak renklendirin!",
     examples=[
-        ["example1.jpg"]  # Örnek görselin dosya yolu
+        ["example_bw.jpg", 35]  # Örnek görsel ve varsayılan render_factor
     ]
 )
 
-# Uygulamayı başlat
 interface.launch()