fixed translation

app.py

@@ -1,9 +1,12 @@
 from diffusers import AudioLDMPipeline
 import torch
 import gradio as gr
-from googletrans import Translator
+from transformers import pipeline
+#from googletrans import Translator
 import os
-print(os.environ)
+
+
+
 if torch.cuda.is_available():
     device = "cuda"
     torch_dtype = torch.float16
@@ -20,21 +23,24 @@ pipe = pipe.to(device)
 def generate_sound(text): 
     print(text)
     # text=translate_text(text)
-    translator = Translator()
-    text=translator.translate(text, src='es',dest="en").text
+    text = translate_text(text)
+    #translator = Translator()
+    #text=translator.translate(text, src='es',dest="en").text
     print(text)
     waveforms = pipe(text, 
-                     num_inference_steps=10, 
+                     num_inference_steps=25, 
                      audio_length_in_s=5,
                      negative_prompt = "low quality, average quality").audios
     rate =16000
     return rate, waveforms[0]
     #return gr.make_waveform((rate, waveforms[0]))
 
+es_en_translator = pipeline("translation",model = "Helsinki-NLP/opus-mt-es-en")
+
+
 def translate_text(text):
-    translator = Translator()
-    translated_text=translator.translate(text, src='es',dest="en")
-    return translated_text.text
+    text = es_en_translator(text)[0].get("translation_text")
+    return text
 
 demo = gr.Blocks()
 with demo:

requirements.txt

@@ -1,4 +1,4 @@
-googletrans==3.1.0a0
+#googletrans==3.1.0a0
 torch
 transformers
 diffusers