Update app.py

app.py

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