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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -52,9 +52,7 @@ if __name__ == "__main__":
52
 
53
  models = [
54
  "text-chem-t5-small-standard",
55
- "text-chem-t5-small-augm",
56
- "text-chem-t5-base-standard",
57
- "text-chem-t5-base-augm",
58
  ]
59
 
60
  metadata_root = pathlib.Path(__file__).parent.joinpath("model_cards")
@@ -71,7 +69,7 @@ if __name__ == "__main__":
71
 
72
  demo = gr.Interface(
73
  fn=run_inference,
74
- title="Text+chem-T5 model",
75
  inputs=[
76
  gr.Dropdown(
77
  models,
 
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")
 
69
 
70
  demo = gr.Interface(
71
  fn=run_inference,
72
+ title="Multitask Text and Chemistry T5",
73
  inputs=[
74
  gr.Dropdown(
75
  models,