Update app.py

app.py

@@ -4,16 +4,20 @@ import gradio as gr
 import torch
 import numpy as np
 from PIL import Image, ImageEnhance
-import spaces
+
+        
+if os.environ.get("HF_TOKEN") or os.environ.get("HF_HOME") or os.environ.get("HUGGINGFACE_HUB_CACHE"):
+    import spaces
+
 
 # Constants
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
 OUTPUT_DIR = "output_minecraft_skins"
 
-print(f"Using {DEVICE}")
+print(f"Using: {DEVICE}")
 
-# Ensure the Minecraft_Skin_Generator repository exists and is up-to-date
+# Repo URL
 REPO_URL = "https://github.com/Nick088Official/Minecraft_Skin_Generator.git"
 REPO_NAME = "Minecraft_Skin_Generator"
 
@@ -31,8 +35,8 @@ else:
 # Change to the repository directory for script execution
 os.chdir(REPO_NAME)
 
-
-@spaces.GPU(duration=75)
+if os.environ.get("HF_TOKEN") or os.environ.get("HF_HOME") or os.environ.get("HUGGINGFACE_HUB_CACHE"):
+    @spaces.GPU(duration=75)
 def run_inference(
     prompt: str,
     stable_diffusion_model: str,
@@ -152,5 +156,3 @@ def create_gradio_ui():
 if __name__ == "__main__":
     demo = create_gradio_ui()
     demo.launch(show_api=False, share=True)
-
-