Update app.py

app.py

@@ -20,8 +20,8 @@ from insightface.data import get_image as ins_get_image
 
 
 device = "cuda"
-# ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors")
-# ckpt_dir_faceid = snapshot_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus")
+ckpt_dir = snapshot_download(repo_id="SunderAli17/SAK")
+ckpt_dir_faceid = snapshot_download(repo_id="SunderAli17/SAK-IP-Adapter-FaceTransform-Plus")
 
 text_encoder = ChatGLMModel.from_pretrained(f'{ckpt_dir}/text_encoder', torch_dtype=torch.float16).half().to(device)
 tokenizer = ChatGLMTokenizer.from_pretrained(f'{ckpt_dir}/text_encoder')
@@ -119,8 +119,8 @@ def infer(prompt,
 
 
 examples = [
-    ["wearing a full suit sitting in a restaurant with candle lights ", "image/image1.png"],
-    ["Cowboy, cowboy hat, Wild Cowboy, background is a western town, cactus, sunset, warm colors, shot with XT4 film, noise, vignette, Kodak film, vintage", "image/image2.png"]
+    ["wearing a full suit sitting in a restaurant with candle lights ", "image/image1.png"]
+    # ["Cowboy, cowboy hat, Wild Cowboy, background is a western town, cactus, sunset, warm colors, shot with XT4 film, noise, vignette, Kodak film, vintage", "image/image2.png"]
 ]