Update app.py

app.py

@@ -14,6 +14,7 @@ from diffusers.utils import load_image
 import torch
 from huggingface_hub import HfApi
 from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 # --- Model Loading ---
 predictor = None