cosmicdream commited on
Commit
c0026e7
1 Parent(s): 2fda7e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
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
- # pil_images = pil_images[0]
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")],