Update app.py

app.py

@@ -9,6 +9,7 @@ import numpy as np
 import os
 import requests
 from tqdm import tqdm
+from safetensors import safe_open
 
 # Initialize Gemini AI
 genai.configure(api_key='YOUR_GEMINI_API_KEY')
@@ -52,8 +53,10 @@ e2tts = E2TTS(
     ),
 )
 
-# Load the pre-trained model
-e2tts.load_state_dict(torch.load(model_path))
+# Load the pre-trained model using safetensors
+with safe_open(model_path, framework="pt", device="cpu") as f:
+    state_dict = {key: f.get_tensor(key) for key in f.keys()}
+    e2tts.load_state_dict(state_dict)
 e2tts.eval()
 
 def generate_podcast_script(content, duration):