akdeniz27 commited on
Commit
4b53411
1 Parent(s): 45b55a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ from predict import run_prediction
5
 
6
  st.set_page_config(layout="wide")
7
 
8
- model_list = ['roberta-base-cuad',
9
- 'roberta-large-cuad',
10
- 'deberta-xlarge-cuad']
11
  st.sidebar.header("Select CUAD Model")
12
  model_checkpoint = st.sidebar.radio("", model_list)
13
 
 
5
 
6
  st.set_page_config(layout="wide")
7
 
8
+ model_list = ['akdeniz27/roberta-base-cuad',
9
+ 'akdeniz27/roberta-large-cuad',
10
+ 'akdeniz27/deberta-xlarge-cuad']
11
  st.sidebar.header("Select CUAD Model")
12
  model_checkpoint = st.sidebar.radio("", model_list)
13