Spaces:

lachine
/

kandinsky-improved

Runtime error

lachine commited on Apr 21, 2023

Commit

0bbd0a0

1 Parent(s): 4ea20ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
 def generate_text(prompt, quality):
     length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}

 import gradio as gr
 from transformers import pipeline
+text_generator = pipeline(model="ai-forever/Kandinsky_2.1")
 def generate_text(prompt, quality):
     length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}