Upload genimage.py

genimage.py

@@ -2,11 +2,12 @@ import spaces
 
 
 def load_pipeline():
-    from diffusers import StableDiffusionXLPipeline
+    from diffusers import AutoPipelineForText2Image
     import torch
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-    pipe = StableDiffusionXLPipeline.from_pretrained(
+    pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
+        custom_pipeline="lpw_stable_diffusion_xl",
         torch_dtype=torch.float16,
     )
     pipe.to(device)
@@ -43,8 +44,8 @@ def generate_image(prompt, neg_prompt):
     }
     try: 
         images = pipe(
-            prompt=prompt + ", masterpiece, best quality, very aesthetic, absurdres",
-            negative_prompt=neg_prompt + ", bad hands, bad foots, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
+            prompt=prompt + ", anime, masterpiece, best quality, very aesthetic, absurdres",
+            negative_prompt=neg_prompt + ", bad hands, bad feet, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
             width=1024,
             height=1024,
             guidance_scale=7.5,