Update run.py

run.py

@@ -1,21 +1,20 @@
-import torch
 from diffusers import StableDiffusionPipeline
+import torch
 
-# Load the Stable Diffusion pipeline
+# Load the model (make sure the model is loaded from the correct path)
 pipeline = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", 
+    "Malik99999/MyModel", 
     torch_dtype=torch.float16
 )
-pipeline.to("cuda")  # Use the GPU
 
-# Load your safetensors weights
-pipeline.unet.load_attn_procs("v1-5-pruned-emaonly.safetensors")
+# Move model to GPU (if available)
+pipeline.to("cuda" if torch.cuda.is_available() else "cpu")
+
+# Example prompt
+prompt = "A beautiful landscape at sunrise"
 
-# Generate an image
-prompt = "A serene landscape with mountains and a lake at sunset"
-result = pipeline(prompt, guidance_scale=7.5)  # Adjust guidance scale if needed
-image = result.images[0]
+# Generate the image
+image = pipeline(prompt).images[0]
 
-# Save and display the image
+# Save or display the image
 image.save("generated_image.png")
-image.show()