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