Update app.py

app.py

@@ -19,6 +19,29 @@ from PIL import Image
 import io
 import torch
 
+from huggingface_hub import hf_hub_download
+import os
+
+# Define the repository and filename
+repo_id = "oguzm/dreamshaper-xl-lightning-dpmsde"
+filename = "dreamshaperXL_lightningDPMSDE.safetensors"
+
+# Download the model
+model_path = hf_hub_download(repo_id=repo_id, filename=filename)
+
+# Ensure the checkpoints directory exists
+checkpoints_dir = "ComfyUI/models/checkpoints"
+os.makedirs(checkpoints_dir, exist_ok=True)
+
+# Define the destination path
+destination_path = os.path.join(checkpoints_dir, filename)
+
+# Copy the model to the checkpoints directory if it doesn't already exist
+if not os.path.exists(destination_path):
+    import shutil
+    shutil.copy(model_path, destination_path)
+
+
 
 URL="http://127.0.0.1"
 OUTPUT_DIR="ComfyUI/output"