Update app.py

app.py

@@ -40,9 +40,9 @@ DEFAULT_STYLE_NAME = "Japanese Anime"
 global models_dict
 use_va = True
 models_dict = {
-#    "Juggernaut": "RunDiffusion/Juggernaut-XL-v8",
-   "RealVision": "SG161222/RealVisXL_V4.0" ,
-   "SDXL":"stabilityai/stable-diffusion-xl-base-1.0" ,
+    "Juggernaut": "RunDiffusion/Juggernaut-XL-v8",
+#   "RealVision": "SG161222/RealVisXL_V4.0" ,
+    "SDXL":"stabilityai/stable-diffusion-xl-base-1.0" ,
 #   "Unstable": "stablediffusionapi/sdxl-unstable-diffusers-y"
 }
 photomaker_path =  hf_hub_download(repo_id="TencentARC/PhotoMaker", filename="photomaker-v1.bin", repo_type="model")
@@ -436,7 +436,7 @@ use_safetensors= False
 # pipe1.scheduler.set_timesteps(50)
 ### 
 pipe2 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
-    models_dict["RealVision"], torch_dtype=torch.float16, use_safetensors=use_safetensors)
+    models_dict["Juggernaut"], torch_dtype=torch.float16, use_safetensors=use_safetensors)
 pipe2 = pipe2.to("cpu")
 pipe2.load_photomaker_adapter(
     os.path.dirname(photomaker_path),
@@ -520,7 +520,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
     num_steps =_num_steps
     use_safe_tensor = True
     if  style_name == "(No style)":
-        sd_model_path = models_dict["RealVision"]
+        sd_model_path = models_dict["Juggernaut"]
     if _model_type == "original":
         pipe = StableDiffusionXLPipeline.from_pretrained(sd_model_path, torch_dtype=torch.float16)
         pipe = pipe.to(device)
@@ -529,7 +529,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
         # pipe.scheduler.set_timesteps(50)
         set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
     elif _model_type == "Photomaker":
-        if _sd_type != "RealVision" and style_name != "(No style)":
+        if _sd_type != "Juggernaut" and style_name != "(No style)":
             pipe = pipe2.to(device)
             pipe.id_encoder.to(device)
             set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)