Update app.py

app.py

@@ -10,11 +10,9 @@ if hf_api_key is None:
 
 # Load model with authentication token
 model_id = "stabilityai/stable-diffusion-3.5-medium"
-pipe = StableDiffusion3Pipeline.from_pretrained(model_id, use_auth_token=hf_api_key)
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium")
 
-# Move model to the correct device
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe.to(device)
+pipe.to("cuda")# If you have GPU access; otherwise, use "cpu"
 
 # Define Gradio interface function
 def generate_image(prompt):