is_space_imported

app.py

@@ -3,12 +3,17 @@ import json
 import torch
 import time
 import random
+
 try:
-    # Only on HuggingFace
     import spaces
     is_space_imported = True
-except ImportError:
+except:
     is_space_imported = False
+    class spaces():
+        def GPU(*args, **kwargs):
+            def decorator(function):
+                return lambda *dummy_args, **dummy_kwargs: function(*dummy_args, **dummy_kwargs)
+            return decorator
 
 from tqdm import tqdm
 from huggingface_hub import snapshot_download
@@ -115,6 +120,7 @@ def update_output(output_format, output_number):
         gr.update(visible = False)
     ]
 
+@spaces.GPU(duration = 420)
 def text2audio(
     prompt,
     output_number,
@@ -150,9 +156,6 @@ def text2audio(
         gr.update(visible = True, value = "Start again to get a different result. The output have been generated in " + ((str(hours) + " h, ") if hours != 0 else "") + ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + str(secondes) + " sec.")
     ]
 
-if is_space_imported:
-    text2audio = spaces.GPU(text2audio, duration = 420)
-
 # Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("""