Update app.py

app.py

@@ -1,14 +1,13 @@
 import os
-if os.environ.get("SPACES_ZERO_GPU") is not None:
-    import spaces
-else:
-    class spaces:
-        @staticmethod
-        def GPU(func):
-            def wrapper(*args, **kwargs):
-                return func(*args, **kwargs)
-            return wrapper
-
+#if os.environ.get("SPACES_ZERO_GPU") is not None:
+import spaces
+#else:
+#    class spaces:
+#        @staticmethod
+#        def GPU(func):
+#            def wrapper(*args, **kwargs):
+#                return func(*args, **kwargs)
+#            return wrapper
 import gradio as gr
 import json
 import logging
@@ -41,10 +40,10 @@ from transformers import CLIPModel, CLIPProcessor, CLIPTextModel, CLIPTokenizer,
 import gc
 import warnings
 model_path = snapshot_download(repo_id="nyanko7/flux-dev-de-distill")
-cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
-os.environ["TRANSFORMERS_CACHE"] = cache_path
-os.environ["HF_HUB_CACHE"] = cache_path
-os.environ["HF_HOME"] = cache_path
+#cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
+#os.environ["TRANSFORMERS_CACHE"] = cache_path
+#os.environ["HF_HUB_CACHE"] = cache_path
+#os.environ["HF_HOME"] = cache_path
 
 device = "cuda" if torch.cuda.is_available() else "cpu"