Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import pipeline
-text_generator = pipeline(model="ai-forever/Kandinsky_2.1")
+text_generator = pipeline(model="ai-forever/Kandinsky_2.1", task="text2img")
 
-def generate_text(prompt, quality):
-    length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}
+def generate_text(prompt, quality="High (Default)"):
+    length_dict = {"Low": 50, "High (Default)": 100, "High": 175}
     length = length_dict[quality]
-    return text_generator(prompt, max_length=length, do_sample=True)[0]['generated_text']
+    return text_generator(prompt, h=768, w=768, num_steps=legnth, batch_size=1, sampler='p_sampler', prior_steps="5", prior_cf_scale=4,)[0]
 
 iface = gr.Interface(
   fn=generate_text,