Update app.py

app.py

@@ -8,7 +8,7 @@ logging.set_verbosity_error()  # This suppresses warnings, including cache migra
 from diffusers import StableDiffusionPipeline
 
 # Load the model
-model_id = "your-username/stable-diffusion-finetuned"
+model_id = "MostafaAly/stable-diffusion-finetuned"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 
 # Check if CUDA is available and move the model to GPU if it is, otherwise use CPU
@@ -18,11 +18,6 @@ pipe.to(device)
 print(f"Using device: {device}")
 
 
-# Load your fine-tuned model from Hugging Face
-model_id = "MostafaAly/stable-diffusion-finetuned"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe.to("cuda")
-
 # Define the function for text-to-image generation
 def generate_image(prompt):
     image = pipe(prompt).images[0]