Update app.py

app.py

@@ -11,9 +11,9 @@ if token:
 else:
     raise ValueError("Hugging Face token not found. Please set it as a repository secret in the Space settings.")
 
-# Load the Stable Diffusion 3.5 model
+# Load the Stable Diffusion 3.5 model with lower precision (float16)
 model_id = "stabilityai/stable-diffusion-3.5-large"
-pipe = StableDiffusion3Pipeline.from_pretrained(model_id)  # Removed torch_dtype argument
+pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16)  # Use float16 precision
 pipe.to("cpu")  # Ensuring it runs on CPU
 
 # Define the path to the LoRA model
@@ -37,10 +37,10 @@ pipe = load_lora_model(pipe, lora_model_path)
 # Function to generate an image from a text prompt
 def generate_image(prompt, seed=None):
     generator = torch.manual_seed(seed) if seed is not None else None
-    image = pipe(prompt, height=1080, width=1080, generator=generator).images[0]
+    # Reduce image size for less memory usage
+    image = pipe(prompt, height=512, width=512, generator=generator).images[0]  # Changed image size
     return image
 
-
 # Gradio interface
 iface = gr.Interface(
     fn=generate_image,
@@ -50,5 +50,4 @@ iface = gr.Interface(
     ],
     outputs="image"
 )
-iface.launch()
-
+iface.launch()