Spaces:
Runtime error
Runtime error
cosmicdream
commited on
Commit
•
c0026e7
1
Parent(s):
2fda7e8
Update app.py
Browse files
app.py
CHANGED
@@ -52,17 +52,12 @@ text = 'радуга на фоне ночного города'
|
|
52 |
# scores += _scores
|
53 |
|
54 |
def model(text, tokenizer=tokenizer, dalle=dalle, vae=vae, top_k=2048, images_num=1, bs=8, top_p=0.995):
|
55 |
-
# _pil_images, _scores = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
56 |
pil_images = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
57 |
-
|
58 |
return pil_images[0]
|
59 |
|
60 |
seed_everything(42)
|
61 |
pil_images = []
|
62 |
-
scores = []
|
63 |
-
top_k = 2048
|
64 |
-
top_p = 0.995
|
65 |
-
images_num = 1
|
66 |
|
67 |
iface = gr.Interface(fn=model,
|
68 |
inputs=[gr.inputs.Textbox(label="Text prompt")],
|
|
|
52 |
# scores += _scores
|
53 |
|
54 |
def model(text, tokenizer=tokenizer, dalle=dalle, vae=vae, top_k=2048, images_num=1, bs=8, top_p=0.995):
|
|
|
55 |
pil_images = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
56 |
+
pil_images = pil_images[0]
|
57 |
return pil_images[0]
|
58 |
|
59 |
seed_everything(42)
|
60 |
pil_images = []
|
|
|
|
|
|
|
|
|
61 |
|
62 |
iface = gr.Interface(fn=model,
|
63 |
inputs=[gr.inputs.Textbox(label="Text prompt")],
|