Update edit_app.py

edit_app.py

@@ -7,7 +7,7 @@ import random
 import gradio as gr
 import torch
 from PIL import Image, ImageOps
-from diffusers import StableDiffusionInstructPix2PixPipeline, StableDiffusionPipeline
+from diffusers import StableDiffusionInstructPix2PixPipeline, StableDiffusionPipeline, DiffusionPipeline
 
 
 help_text = """
@@ -50,7 +50,9 @@ auth_token = os.environ['auth_token']
 model_id = "deleom/giornalettismoai"
 modelid = "CompVis/stable-diffusion-v1-4"
 device = "cuda"
-stable_diffusion_model = StableDiffusionPipeline.from_pretrained(modelid, revision="fp16", torch_dtype=torch.float16, use_auth_token=auth_token) 
+#stable_diffusion_model = StableDiffusionPipeline.from_pretrained(modelid, revision="fp16", torch_dtype=torch.float16, use_auth_token=auth_token) 
+stable_diffusion_model = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+pipe.to("cuda")
 stable_diffusion_model.to(device)
 
 def main():