Update app.py

app.py

@@ -3,6 +3,14 @@ import torch
 from huggingface_hub import hf_hub_download
 from diffusers import DiffusionPipeline
 from cog_sdxl.dataset_and_utils import TokenEmbeddingsHandler
+import os
+import git
+
+# Clone the cog-sdxl repository if it doesn't exist
+repo_url = "https://github.com/replicate/cog-sdxl.git"
+repo_dir = "./cog-sdxl"
+if not os.path.exists(repo_dir):
+    git.Repo.clone_from(repo_url, repo_dir)
 
 # Load the model pipeline
 pipe = DiffusionPipeline.from_pretrained(
@@ -11,6 +19,7 @@ pipe = DiffusionPipeline.from_pretrained(
     variant="fp16",
 ).to("cuda")
 
+# Load the LoRA weights
 pipe.load_lora_weights("fofr/sdxl-emoji", weight_name="lora.safetensors")
 
 # Load token embeddings