Update app.py

app.py

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