Update app.py

app.py

@@ -10,17 +10,17 @@ from typing import List
 import shutil
 import numpy as np
 import random
-
+import spaces
 # Ensure `spaces` is imported first
-try:
-    import spaces
-except ImportError:
-    class spaces:
-        @staticmethod
-        def GPU(func=None, duration=None):
-            def wrapper(fn):
-                return fn
-            return wrapper if func is None else wrapper(func)
+#try:
+#    import spaces
+#except ImportError:
+#    class spaces:
+#        @staticmethod
+#        def GPU(func=None, duration=None):
+#            def wrapper(fn):
+#                return fn
+#            return wrapper if func is None else wrapper(func)
 
 # Now import CUDA-related libraries
 import torch
@@ -113,7 +113,7 @@ def check_gpu_availability():
     else:
         print("CUDA is not available. Running on CPU.")
 
-check_gpu_availability()
+#check_gpu_availability()
 
 #@spaces.GPU()
 def generate_image_with_flux_old(