vasudevgupta commited on
Commit
8f330dd
1 Parent(s): 957bbf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ biggan.to(device).eval()
29
  text_model.to(device).eval()
30
  autoencoder.to(device).eval()
31
 
32
- def get_image(text_query):
33
  return generate_image_from_text(text_query, text_tokenizer, text_model, autoencoder, biggan, device=device)
34
 
35
- gr.Interface(fn=get_image, inputs="text", outputs="image").launch()
 
29
  text_model.to(device).eval()
30
  autoencoder.to(device).eval()
31
 
32
+ def generate_image(text_query):
33
  return generate_image_from_text(text_query, text_tokenizer, text_model, autoencoder, biggan, device=device)
34
 
35
+ gr.Interface(fn=generate_image, inputs="text", outputs="image").launch()