Update app.py

app.py

@@ -1,9 +1,8 @@
-# Stable Diffusion Hugging Face App (Turbo Version for Fast Startup)
+# Stable Diffusion Hugging Face App (Turbo Version with Fixes)
 
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline, DDIMScheduler
-from transformers import CLIPTextModel, CLIPTokenizer
 
 # Load the lightweight Stable Diffusion Turbo model
 model_id = "stabilityai/sd-turbo"
@@ -35,9 +34,9 @@ def custom_loss_placeholder(image_tensor):
 # Generate image based on prompt and style
 
 def generate(prompt, style, seed):
-    torch.manual_seed(seed)
+    generator = torch.manual_seed(seed)
     full_prompt = f"{prompt}, {STYLE_MAP.get(style, '')}"
-    result = pipe(full_prompt, guidance_scale=7.5).images[0]
+    result = pipe(full_prompt, guidance_scale=7.5, generator=generator).images[0]
     return result
 
 # Gradio UI