Update app.py

app.py

@@ -1,17 +1,23 @@
+import os
+import git
 import gradio as gr
 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
+import subprocess
 
 # 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):
+    print("Cloning cog-sdxl repository...")
     git.Repo.clone_from(repo_url, repo_dir)
 
+# Install cog-sdxl as a local package
+print("Installing cog-sdxl package...")
+subprocess.check_call([sys.executable, "-m", "pip", "install", "-e", repo_dir])
+
 # Load the model pipeline
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",