Update main_app.py

main_app.py

@@ -75,6 +75,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
     return p.replace("{prompt}", positive), n + negative
 
 DESCRIPTIONs = """ㅤㅤㅤ """
+
 DESCRIPTION = """ㅤㅤㅤ """
 
 if not torch.cuda.is_available():
@@ -109,14 +110,30 @@ if torch.cuda.is_available():
         pipe.enable_model_cpu_offload()
         pipe2.enable_model_cpu_offload()
     else:
-        pipe.to(device)
-        pipe2.to(device)
+        pipe.to(device)    
+        pipe2.to(device)    
         print("Loaded on Device!")
     
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
+else:
+    pipe = DiffusionPipeline.from_pretrained(
+        "SG161222/RealVisXL_V4.0",
+        torch_dtype=torch.float32,
+        use_safetensors=True,
+        add_watermarker=False
+    )
+    pipe2 = DiffusionPipeline.from_pretrained(
+        "SG161222/RealVisXL_V3.0",
+        torch_dtype=torch.float32,
+        use_safetensors=True,
+        add_watermarker=False
+    )
+    pipe.to(device)
+    pipe2.to(device)
+    print("Loaded on Device!")
 
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"