Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from camb_ai_mars5_tts import Mars5TTS, InferenceConfig
+from inference import Mars5TTS, InferenceConfig
 import librosa
 import torch
 import numpy as np
@@ -9,8 +9,11 @@ if torch.cuda.is_available():
     torch.cuda.empty_cache()
 
 # MARS5 TTS 모델 로드
-mars5 = Mars5TTS.from_pretrained("CAMB-AI/MARS5-TTS")
-config = InferenceConfig(temperature=0.7)  # Prosody 제어를 위한 온도 설정
+try:
+    mars5 = Mars5TTS.from_pretrained("CAMB-AI/MARS5-TTS")
+    config = InferenceConfig(temperature=0.7)  # Prosody 제어를 위한 온도 설정
+except Exception as e:
+    print(f"Model loading error: {str(e)}")
 
 def clone_with_prosody(text, ref_audio, enhance_prosody=True):
     """