Update run.py

run.py

@@ -1,20 +1,13 @@
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Load the model (make sure the model is loaded from the correct path)
-pipeline = StableDiffusionPipeline.from_pretrained(
-    "Malik99999/MyModel", 
-    torch_dtype=torch.float16
-)
-
-# Move model to GPU (if available)
+# Load the model from Hugging Face repo
+pipeline = StableDiffusionPipeline.from_pretrained("Malik99999/MyModel")
 pipeline.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Example prompt
-prompt = "A beautiful landscape at sunrise"
-
-# Generate the image
+# Generate an image from a prompt
+prompt = "A beautiful landscape at sunset"
 image = pipeline(prompt).images[0]
 
-# Save or display the image
+# Save the generated image
 image.save("generated_image.png")