Update app.py

app.py

@@ -10,9 +10,12 @@ from threading import Thread
 from transformers import pipeline
 import emoji
 
-ru2en = pipeline("translation", model="Helsinki-NLP/opus-mt-ru-en")
-def translate(text):
-  return ru2en(text)[0]["translation_text"]
+ru2en = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-ru-en")
+# def translate(text):
+  # return ru2en(text)[0]["translation_text"]
+    
+# def translate(inputs, noise_level, ru2en=ru2en):
+#   return ru2en(prompt_with_noise)#[0]["translation_text"]
 
 text_gen=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
 def get_prompts(prompt_text):
@@ -111,7 +114,7 @@ with gr.Blocks(css='style.css') as demo:
                 run4 = gr.Button("Generate").style(full_width=False)
                 output4=gr.Image()
         
-        Translate.click(translate, inputs=[input_text_ru], outputs=[input_text], queue=False)
+        Translate.click(ru2en, inputs=[input_text_ru], outputs=[input_text], queue=False)
         see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
         run1.click(send_it, inputs=[prompt, noise_level], outputs=[output1])
         run2.click(send_it, inputs=[prompt, noise_level], outputs=[output2])