Update app.py

app.py

@@ -3,8 +3,7 @@ import random
 
 import gradio as gr
 import numpy as np
-import PIL.Image
-from PIL import Image 
+import PIL.Image 
 import torch
 import torchvision.transforms.functional as TF
 from diffusers import (
@@ -113,7 +112,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 
 def run(
-    image: Image,
+    image: PIL.Image.Image,
     prompt: str,
     negative_prompt: str,
     style_name: str = DEFAULT_STYLE_NAME,
@@ -124,7 +123,7 @@ def run(
     seed: int = 0,
     progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
-    image = Image.convert("RGB")
+    image = image.convert("RGB")
     image = TF.to_tensor(image) > 0.5
     image = TF.to_pil_image(image.to(torch.float32))