christofid commited on
Commit
db8b533
1 Parent(s): a668117

Add t5-base variants

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -52,7 +52,9 @@ if __name__ == "__main__":
52
 
53
  models = [
54
  "multitask-text-and-chemistry-t5-small-standard",
55
- "multitask-text-and-chemistry-t5-small-augm"
 
 
56
  ]
57
 
58
  metadata_root = pathlib.Path(__file__).parent.joinpath("model_cards")
 
52
 
53
  models = [
54
  "multitask-text-and-chemistry-t5-small-standard",
55
+ "multitask-text-and-chemistry-t5-small-augm",
56
+ "multitask-text-and-chemistry-t5-base-standard",
57
+ "multitask-text-and-chemistry-t5-base-augm",
58
  ]
59
 
60
  metadata_root = pathlib.Path(__file__).parent.joinpath("model_cards")