Update app.py

app.py

@@ -13,7 +13,7 @@ from functools import partial
 from gradio.components import Image, Textbox
 import gradio as gr
 
-from PIL import Image
+from PIL import Image as PILImage
 from huggingface_hub import snapshot_download
 import os
 import random
@@ -67,9 +67,7 @@ def generate_img(
     seed=0,
     randomize_seed=False,
 ):
-    # color_converted_image = cv2.cvtColor(np_image, cv2.COLOR_BGR2RGB)
-
-    pil_image = Image.fromarray(np_image)
+    pil_image = PILImage.fromarray(np_image)
 
     seed = int(randomize_seed_fn(seed, randomize_seed))
     set_env(seed)