christofid commited on
Commit
8cca560
1 Parent(s): 0ef0026

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,8 +51,8 @@ def run_inference(
51
  if __name__ == "__main__":
52
 
53
  models = [
54
- "text-chem-t5-small-standard",
55
- "text-chem-t5-small-augm"
56
  ]
57
 
58
  metadata_root = pathlib.Path(__file__).parent.joinpath("model_cards")
@@ -74,7 +74,7 @@ if __name__ == "__main__":
74
  gr.Dropdown(
75
  models,
76
  label="Language model",
77
- value="text-chem-t5-base-augm",
78
  ),
79
  gr.Radio(
80
  choices=[
 
51
  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")
 
74
  gr.Dropdown(
75
  models,
76
  label="Language model",
77
+ value="multitask-text-and-chemistry-t5-small-augm",
78
  ),
79
  gr.Radio(
80
  choices=[