Update app.py

app.py

@@ -21,7 +21,7 @@ MAX_SEED = np.iinfo(np.int32).max
 
 
 # Initialize the pipeline and download the sd3 medium model
-pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
 pipe.to(device)
 
 # superprompt-v1
@@ -30,7 +30,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
 model.to(device)
 
 # Define the image generation function
-@spaces.GPU(duration=60 * 2)
+@spaces.GPU(duration=90)
 def generate_image(prompt, enhance_prompt, negative_prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt):
     if seed == 0:
         seed = random.randint(1, 2**32-1)