Update app.py

app.py

@@ -165,7 +165,7 @@ def translate_local(text_to_translate, target_lang='ta-IN', device=None):
 
     return translated_text
 
-def synthesize_speech(synth_text, target_lang, gender="Male", pace="normal", output_path="temp_audio_synthesized.wav", device=None):
+def synthesize_speech(synth_text, target_lang, gender="Male", pace="normal", output_path="temp_audio_synthesized.wav", device="cpu"):
 
     ref_audio_path = str('reference/'+target_lang.upper()+'/'+gender.upper()+'_'+pace.upper()+'.wav')
     ref_text_path = str('reference/'+target_lang.upper()+'/'+gender.upper()+'_'+pace.upper()+'.txt')
@@ -177,7 +177,7 @@ def synthesize_speech(synth_text, target_lang, gender="Male", pace="normal", out
     print("> Loading IndicF5 TTS model (ai4bharat/IndicF5)...")
     indicf5_repo_id = "ai4bharat/IndicF5"
     token = os.environ.get("HF_TOKEN")
-    tts_model = AutoModel.from_pretrained(indicf5_repo_id, token=token, trust_remote_code=True).to(device)
+    tts_model = AutoModel.from_pretrained(indicf5_repo_id, token=token, trust_remote_code=True)
     
     audio = tts_model(synth_text, ref_audio_path=ref_audio_path, ref_text=ref_text)