Update app.py

app.py

@@ -13,8 +13,6 @@ import torch
 from PIL import Image
 from io import BytesIO
 from diffusers import StableDiffusionImg2ImgPipeline, AutoencoderKL, DiffusionPipeline
-from diffusers.utils import load_image
-from safety_checker import StableDiffusionSafetyChecker
 
 DESCRIPTION = "# SDXL"
 if not torch.cuda.is_available():
@@ -64,7 +62,6 @@ def generate(
     if torch.cuda.is_available():
 
         if not use_vae:
-            safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
             
         if use_vae:
@@ -76,7 +73,7 @@ def generate(
             pipe.fuse_lora(lora_scale=0.7)
             
         response = requests.get("https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg")
-        init_image = Image.open(BytesIO(response.content)).convert("RGB")
+        init_image = Image.open(img)
         init_image = init_image.resize((1024, 1024))
 
         if ENABLE_CPU_OFFLOAD: