Spaces:
Runtime error
Runtime error
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")],
|