import torch from diffusers import StableDiffusionPipeline def generate_and_save_image(prompt, output_filename="generated_image.png"): print("AI Model load ho raha hai, bara-e-maharbani intizar karein...") # Stable Diffusion ka free model load karna model_id = "runwayml/stable-diffusion-v1-5" # Agar aap ke paas GPU (Nvidia) hai to yeh fast chalega, warna CPU par chale gaa device = "cuda" if torch.cuda.is_available() else "cpu" # Model ko initialize karna pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) pipe = pipe.to(device) print(f"Prompt par kaam ho raha hai: '{prompt}'") # Image generate karna image = pipe(prompt).images[0] # Image ko save karna image.save(output_filename) print(f"Mubarak ho! Aap ki picture '{output_filename}' ke naam se save ho gayi hai.") # --- Code ko chalane ka ...