update

app.py

@@ -6,7 +6,7 @@ from typing import List, Tuple
 import gradio as gr
 import torch
 import torchaudio
-import spaces
+# import spaces
 from dataclasses import dataclass
 from generator import Segment, load_csm_1b
 from huggingface_hub import login
@@ -144,7 +144,7 @@ def generate_speech(
         
         progress(1.0, "Completed!")
         return output_path
-    except spaces.zero.gradio.HTMLError as e:
+    except Exception as e:
         # Handle ZeroGPU quota exceeded error
         error_message = str(e)
         if "GPU quota exceeded" in error_message:

requirements.txt

@@ -7,3 +7,4 @@ moshi==0.2.2
 torchtune==0.4.0
 torchao==0.9.0
 silentcipher @ git+https://github.com/SesameAILabs/silentcipher@master
+gradio==4.44.1

test_model.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import torchaudio
-import spaces
+# import spaces
 from generator import Segment, load_csm_1b
 from huggingface_hub import login
 
@@ -22,7 +22,7 @@ def login_huggingface():
         print("No token provided. Some models may not be accessible.")
         return False
 
-@spaces.GPU
+# @spaces.GPU
 def generate_test_audio(text, speaker_id, device):
     """Generate test audio using ZeroGPU"""
     generator = load_csm_1b(device=device)