Update app.py

app.py

@@ -612,7 +612,6 @@ def install_parler_tts():
 # Call the function to install parler-tts
 install_parler_tts()
 
-
 import gradio as gr
 import requests
 import os
@@ -943,7 +942,7 @@ def generate_map(location_names):
         if geocode_result:
             location = geocode_result[0]['geometry']['location']
             folium.Marker(
-                [location['lat'], location['lng']],
+                [location['lat'], 'lng']],
                 tooltip=f"{geocode_result[0]['formatted_address']}"
             ).add_to(m)
     
@@ -1154,7 +1153,14 @@ def process_chunk(chunk, model, tokenizer, device):
 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)
 
     text_chunks = chunk_text(text)
@@ -1259,3 +1265,4 @@ demo.launch(share=True)
 
 
 
+