Update app.py

app.py

@@ -7,15 +7,16 @@ from os.path import getsize
 from pathlib import Path
 from importlib.metadata import version, PackageNotFoundError
 
-try:
+import gradio as gr
+from gradio.utils import get_space
+
+space_id = get_space()
+
+if space_id:
     import spaces  # it's for ZeroGPU
-except ImportError:
-    ...
 
 import torch
 import torchaudio
-import gradio as gr
-
 
 from huggingface_hub import hf_hub_download
 
@@ -179,7 +180,7 @@ tech_libraries = f"""
 - vocos: {version("vocos")}
 - gradio: {version("gradio")}
 - huggingface_hub: {version("huggingface_hub")}
-- spaces: {spaces_version}
+- spaces: {spaces_version}, id: {space_id}
 - torch: {version("torch")}
 - torchaudio: {version("torchaudio")}
 - scipy: {version("scipy")}
@@ -315,14 +316,9 @@ def inference(
     return [gr.Audio("audio.wav"), rtf_value]
 
 
-try:
-    if hasattr(spaces, "GPU"):
-        inference_func = spaces.GPU(inference)
-    else:
-        inference_func = inference
-except Exception as e:
-    print(f"An error occurred during ZeroGPU setup: {e}")
-    inference_func = inference
+inference_func = inference
+if space_id:
+    inference_func = spaces.GPU(inference)
 
 
 demo = gr.Blocks(