jruneofficial commited on
Commit
01210e1
1 Parent(s): 2d0cabb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from pytorch_pretrained_biggan import BigGAN, truncated_noise_sample, one_hot_fr
16
  config = {
17
  "model_name": "smangrul/Multimodal-Challenge",
18
  "base_model_name": "distilbert-base-uncased",
19
- "image_gen_model": "biggan-deep-128",
20
  "max_length": 20,
21
  "freeze_text_model": True,
22
  "freeze_image_gen_model": True,
 
16
  config = {
17
  "model_name": "smangrul/Multimodal-Challenge",
18
  "base_model_name": "distilbert-base-uncased",
19
+ "image_gen_model": "biggan-deep-512",
20
  "max_length": 20,
21
  "freeze_text_model": True,
22
  "freeze_image_gen_model": True,