Update app.py

app.py

@@ -1,23 +1,27 @@
 import os
 import git
+import subprocess
+import sys
 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 subprocess
 
-# Clone the cog-sdxl repository if it doesn't exist
+# Check if the cog-sdxl repository exists, otherwise clone it
 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
+# Install the cog-sdxl repository locally
 print("Installing cog-sdxl package...")
 subprocess.check_call([sys.executable, "-m", "pip", "install", "-e", repo_dir])
 
+# Now import the required module after it has been installed
+from cog_sdxl.dataset_and_utils import TokenEmbeddingsHandler
+
 # Load the model pipeline
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",