Update app.py

app.py

@@ -10,6 +10,7 @@ import scipy.io.wavfile
 import requests
 from dotenv import load_dotenv
 import deepgram
+import torch
 
 # Load environment variables
 load_dotenv()
@@ -18,6 +19,11 @@ load_dotenv()
 DEEPGRAM_API_KEY = os.getenv("DEEPGRAM_API_KEY")
 
 # Load Bark TTS models
+
+
+# Modify this part in the Bark generation code where the model is loaded
+checkpoint = torch.load(ckpt_path, map_location=device, weights_only=False)
+
 preload_models()
 
 # Conversation history