sagawa commited on
Commit
554cc90
·
1 Parent(s): 5d320ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import streamlit as st
16
 
17
  class CFG():
18
  input_data = st.text_area('enter chemical reaction (e.g. REACTANT:NCCO.O=C1COCC(=O)O1CATALYST: REAGENT: SOLVENT:c1ccncc1)')
19
- model_name_or_path = 'sagawa/ZINC-t5'
20
  model = 't5'
21
  num_beams = 5
22
  num_return_sequences = 5
 
16
 
17
  class CFG():
18
  input_data = st.text_area('enter chemical reaction (e.g. REACTANT:NCCO.O=C1COCC(=O)O1CATALYST: REAGENT: SOLVENT:c1ccncc1)')
19
+ model_name_or_path = 'sagawa/ZINC-t5-productpredicition'
20
  model = 't5'
21
  num_beams = 5
22
  num_return_sequences = 5