temporary API rate limit fix

app.py

@@ -2,14 +2,19 @@ import random
 import torch
 import gradio as gr
 from gradio.mix import Series
+from transformers import pipeline
 from rudalle.pipelines import generate_images
 from rudalle import get_rudalle_model, get_tokenizer, get_vae
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+translation_pipe = pipeline("translation", model="facebook/wmt19-en-ru", device=0)
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae().to(device)
 
+def translation_wrapper(text: str):
+    return translation_pipe(text)[0]["translation_text"]
+
 def dalle_wrapper(prompt: str):
     top_k, top_p = random.choice([
         (1024, 0.98),
@@ -30,8 +35,9 @@ def dalle_wrapper(prompt: str):
     return title, images[0]
 
 
-translator = gr.Interface.load("huggingface/facebook/wmt19-en-ru", 
-                               inputs=[gr.inputs.Textbox(label="What would you like to see?")])
+translator = gr.Interface(fn=translation_wrapper, 
+                          inputs=[gr.inputs.Textbox(label='What would you like to see?')],
+                          outputs="text")
 outputs = [
     gr.outputs.HTML(label=""),   
     gr.outputs.Image(label=""),