RC1

app.py

@@ -65,10 +65,10 @@ def run(prompt, aspect_ratio, model):
     
     rudalle_ar = RuDalleAspectRatio(
         dalle=dalle, vae=vae, tokenizer=tokenizer,
-        aspect_ratio=aspect_ratio_value, bs=2, device=device
+        aspect_ratio=aspect_ratio_value, bs=1, device=device
     )
     
-    _, result_pil_images = rudalle_ar.generate_images(prompt, top_k, 0.975, 2)
+    _, result_pil_images = rudalle_ar.generate_images(prompt, top_k, 0.975, 1)
     #np_images = map(image_to_np,result_pil_images)
     #np_grid = np_gallery(np.array(list(np_images)),2)
     #result_grid = Image.fromarray(np_grid)
@@ -76,7 +76,7 @@ def run(prompt, aspect_ratio, model):
     return(result_pil_images[0], None)
 
 image = gr.outputs.Image(type="pil", label="Your result")
-css = ".output-image{height: 528px !important}"
+css = ".output-image{height: 260px !important}"
 iface = gr.Interface(fn=run, inputs=[
     gr.inputs.Textbox(label="Prompt (if not in Russian, it will be automatically translated to Russian)",default="an amusement park in mars"),
     #gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=45,maximum=50,minimum=1,step=1),

requirements.txt

@@ -4,4 +4,4 @@ gitpython
 rudalle==1.0.0
 einops
 numpy
-googletrans
+googletrans==3.1.0a0