Update app.py

app.py

@@ -348,7 +348,7 @@ def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         results = response.json().get("news_results", [])
         news_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Birmingham Today</h2>
@@ -509,7 +509,12 @@ def generate_audio_elevenlabs(text):
 def generate_audio_parler_tts(text):
     model_id = 'parler-tts/parler_tts_mini_v0.1'
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
-    model = ParlerTTSForConditionalGeneration.from_pretrained(model_id).to(device)
+    try:
+        model = ParlerTTSForConditionalGeneration.from_pretrained(model_id).to(device)
+    except torch.cuda.OutOfMemoryError:
+        print("CUDA out of memory. Switching to CPU.")
+        device = "cpu"
+        model = ParlerTTSForConditionalGeneration.from_pretrained(model_id).to(device)
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 
     description = "A female speaker with a slightly low-pitched voice delivers her words quite expressively, in a very confined sounding environment with clear audio quality. She speaks very fast."