Update app.py

app.py

@@ -2,39 +2,30 @@ import gradio as gr
 import cv2
 import numpy as np
 from PIL import Image
-import io
 
 def convert_to_sketch(image):
-    img = np.array(image)
+  img = np.array(image)
 
-    # Ubah gambar ke grayscale
-    gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+  #ubah gambar ke grayscale
+  gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 
-    # Inversi warna
-    inverted_image = cv2.bitwise_not(gray_image)
+  #inversi warna
+  inverted_image = cv2.bitwise_not(gray_image)
 
-    # Terapkan gaussian blur
-    blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
+  #terapkan gaussian blur
+  blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
 
-    # Inversi hasil blur
-    inverted_blurred = cv2.bitwise_not(blurred)
+  #inversi hasil blur
+  inverted_blurred = cv2.bitwise_not(blurred)
 
-    # Buat sketsa
-    sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
+  #buat sketsa
+  sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
 
-    # Konversi sketsa menjadi gambar PIL
-    pil_image = Image.fromarray(sketch)
-    
-    # Menyimpan gambar dalam format JPG
-    img_byte_arr = io.BytesIO()
-    pil_image.save(img_byte_arr, format="JPEG")
-    img_byte_arr.seek(0)  # Kembali ke awal stream
-    
-    return img_byte_arr  # Gradio akan menangani ini sebagai file gambar
+  return sketch
 
-# Interface
+#interface
 gr.Interface(fn=convert_to_sketch,
             inputs="image",
-            outputs=gr.Image(type="pil"),
+            outputs="image",
             title="Konversi gambar ke sketsa",
-            description="Unggah gambar dan dapatkan versi sketsa dalam format JPG").launch()
+            description="Unggah gambar dan dapatkan versi sketsa").launch()