Update app.py

app.py

@@ -11,14 +11,16 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
 
 repo = "dataautogpt3/OpenDalleV1.1"
+repo = "stabilityai/stable-diffusion-3-medium"
+
 # pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
-pipeline = AutoPipelineForText2Image.from_pretrained('dataautogpt3/OpenDalleV1.1', torch_dtype=torch.float16).to('cuda')        
+pipeline = AutoPipelineForText2Image.from_pretrained(repo, torch_dtype=torch.float16).to('cuda')        
 
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1344
 
-@spaces.GPU(duration=150)
+@spaces.GPU(duration=160)
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed: