Spaces:

jruneofficial
/

text2imgtest

Runtime error

jruneofficial commited on Feb 19, 2022

Commit

e5241f9

1 Parent(s): 01210e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from PIL import Image
 from pytorch_pretrained_biggan import BigGAN, truncated_noise_sample, one_hot_from_names, one_hot_from_int
 config = {
-    "model_name": "smangrul/Multimodal-Challenge",
     "base_model_name": "distilbert-base-uncased",
     "image_gen_model": "biggan-deep-512",
     "max_length": 20,

 from pytorch_pretrained_biggan import BigGAN, truncated_noise_sample, one_hot_from_names, one_hot_from_int
 config = {
+    "model_name": "keras-io/multimodal-entailment",
     "base_model_name": "distilbert-base-uncased",
     "image_gen_model": "biggan-deep-512",
     "max_length": 20,