Update app.py

app.py

@@ -75,17 +75,17 @@ torch._inductor.config.epilogue_fusion = False
 torch._inductor.config.coordinate_descent_check_all_directions = True
 
 # Ensure model and scheduler are initialized in GPU-enabled function
-if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(
-        repo, 
-        tokenizer_3=tokenizer_3, 
-        text_encoder_3=text_encoder_3, 
-        torch_dtype=torch.float16).to("cuda")
-    pipe2 = StableDiffusion3Img2ImgPipeline.from_pretrained(
-        repo, 
-        tokenizer_3=tokenizer_3, 
-        text_encoder_3=text_encoder_3, 
-        torch_dtype=torch.float16).to("cuda")
+#if torch.cuda.is_available():
+pipe = StableDiffusion3Pipeline.from_pretrained(
+    repo, 
+    tokenizer_3=tokenizer_3, 
+    text_encoder_3=text_encoder_3, 
+    torch_dtype=torch.float16).to("cuda")
+pipe2 = StableDiffusion3Img2ImgPipeline.from_pretrained(
+    repo, 
+    tokenizer_3=tokenizer_3, 
+    text_encoder_3=text_encoder_3, 
+    torch_dtype=torch.float16).to("cuda")
 
 pipe.set_progress_bar_config(disable=True)