Update app.py

app.py

@@ -1,4 +1,3 @@
-#path1.0398
 import os
 import random
 import uuid
@@ -9,8 +8,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import DiffusionPipeline
-from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, FlowMatchEulerDiscreteScheduler
+from diffusers import StableDiffusion3Pipeline, FlowMatchEulerDiscreteScheduler
 from typing import Tuple
 
 # BaseConditions
@@ -46,56 +44,46 @@ style_list = [
 ]
 
 collage_style_list = [
-
-
     {
         "name": "B & W",
         "prompt": "black and white collage of {prompt}. monochromatic, timeless, classic, dramatic contrast",
         "negative_prompt": "colorful, vibrant, bright, flashy",
     },
-
     {
         "name": "Polaroid",
         "prompt": "collage of polaroid photos featuring {prompt}. vintage style, high contrast, nostalgic, instant film aesthetic",
         "negative_prompt": "digital, modern, low quality, blurry",
     },
-    
     {
         "name": "Watercolor",
         "prompt": "watercolor collage of {prompt}. soft edges, translucent colors, painterly effects",
         "negative_prompt": "digital, sharp lines, solid colors",
     },
-
     {
         "name": "Cinematic",
         "prompt": "cinematic collage of {prompt}. film stills, movie posters, dramatic lighting",
         "negative_prompt": "static, lifeless, mundane",
     },
-    
     {
         "name": "Nostalgic",
         "prompt": "nostalgic collage of {prompt}. retro imagery, vintage objects, sentimental journey",
         "negative_prompt": "contemporary, futuristic, forward-looking",
     },
-
     {
         "name": "Vintage",
         "prompt": "vintage collage of {prompt}. aged paper, sepia tones, retro imagery, antique vibes",
         "negative_prompt": "modern, contemporary, futuristic, high-tech",
     },
-    
     {
         "name": "Scrapbook",
         "prompt": "scrapbook style collage of {prompt}. mixed media, hand-cut elements, textures, paper, stickers, doodles",
         "negative_prompt": "clean, digital, modern, low quality",
     },
-    
     {
         "name": "NeoNGlow",
         "prompt": "neon glow collage of {prompt}. vibrant colors, glowing effects, futuristic vibes",
         "negative_prompt": "dull, muted colors, vintage, retro",
     },
-    
     {
         "name": "Geometric",
         "prompt": "geometric collage of {prompt}. abstract shapes, colorful, sharp edges, modern design, high quality",
@@ -106,14 +94,11 @@ collage_style_list = [
         "prompt": "thematic collage of {prompt}. cohesive theme, well-organized, matching colors, creative layout",
         "negative_prompt": "random, messy, unorganized, clashing colors",
     },
-
     {
         "name": "Retro Pop",
         "prompt": "retro pop art collage of {prompt}. bold colors, comic book style, halftone dots, vintage ads",
         "negative_prompt": "subdued colors, minimalist, modern, subtle",
     },
-
-
     {
         "name": "No Style",
         "prompt": "{prompt}",
@@ -153,12 +138,11 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 if torch.cuda.is_available():
-    pipe = DiffusionPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-3-medium",
+    repo = "stabilityai/stable-diffusion-3-medium"  # You might want to replace with your specific repo if different
+    pipe = StableDiffusion3Pipeline.from_pretrained(
+        repo,
         torch_dtype=torch.float16,
-        use_safetensors=True,
-        add_watermarker=False,
-        variant="fp16"
+        revision="refs/pr/26"
     ).to(device)
 
     if ENABLE_CPU_OFFLOAD:
@@ -248,10 +232,8 @@ def generate(
     return [unique_name], seed
 
 examples = [
-
     "Portrait of a beautiful woman in a hat, summer outfit, with freckles on her face, in a close up shot, with sunlight, outdoors, in soft light, with a beach background, looking at the camera, with high resolution photography, in the style of Hasselblad X2D50c --ar 85:128 --v 6.0 --style raw",
     "Flying food photography with [Two Burgers] as the main subject, Splashes of Toppings and Seasonings, [Rocket Lettuce], [Cheddar Flavored Cheese], [Onion], [Pickles], [Special Sauce], [Sesame Bun], [ sea salt crystals] ::3 Capturing the dynamic splashes of food using high-speed photography , photorealistic, surrealism style, [white background], trending background [clean], Minimalist ::2 [Cuware], [Table], [ Steam], [Smoke], [Vegetable Leaves], [Tomato] ::-0.5 Ad Posters, Pro-Grade Color Grading, Studio Lighting, Rim Lights, [Layered Comps], EOS-1D X Mark III, 500px, Behance, concept art"
-    
 ]
 
 css = '''
@@ -407,4 +389,24 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()
+
+# Updated inference function
+@spaces.GPU
+def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+        
+    generator = torch.Generator().manual_seed(seed)
+    
+    image = pipe(
+        prompt = prompt, 
+        negative_prompt = negative_prompt,
+        guidance_scale = guidance_scale, 
+        num_inference_steps = num_inference_steps, 
+        width = width, 
+        height = height,
+        generator = generator
+    ).images[0] 
+    
+    return image, seed