Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
-
text_generator = pipeline(model="ai-forever/Kandinsky_2.1")
|
4 |
|
5 |
-
def generate_text(prompt, quality):
|
6 |
-
length_dict = {"Low":
|
7 |
length = length_dict[quality]
|
8 |
-
return text_generator(prompt,
|
9 |
|
10 |
iface = gr.Interface(
|
11 |
fn=generate_text,
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
+
text_generator = pipeline(model="ai-forever/Kandinsky_2.1", task="text2img")
|
4 |
|
5 |
+
def generate_text(prompt, quality="High (Default)"):
|
6 |
+
length_dict = {"Low": 50, "High (Default)": 100, "High": 175}
|
7 |
length = length_dict[quality]
|
8 |
+
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]
|
9 |
|
10 |
iface = gr.Interface(
|
11 |
fn=generate_text,
|