Update app.py

app.py

@@ -20,7 +20,7 @@ import shutil
 import numpy as np
 import random
 from diffusers import DiffusionPipeline
-
+import spaces
 # Initialize FLUX pipeline outside of the function
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -34,15 +34,15 @@ nltk.download('punkt')
 multiprocessing.set_start_method("spawn", force=True)
 
 # GPU Fallback Setup
-if os.environ.get("SPACES_ZERO_GPU") is not None:
-    import spaces
-else:
-    class spaces:
-        @staticmethod
-        def GPU(func=None, duration=None):
-            def wrapper(fn):
-                return fn
-            return wrapper if func is None else wrapper(func)
+#if os.environ.get("SPACES_ZERO_GPU") is not None:
+#    import spaces
+#else:
+#    class spaces:
+#        @staticmethod
+#        def GPU(func=None, duration=None):
+##            def wrapper(fn):
+#                return fn
+#            return wrapper if func is None else wrapper(func)
 
 # Download necessary NLTK data
 def setup_nltk():