Spaces:
Runtime error
Runtime error
prateekagrawal
commited on
Commit
•
aeb3532
1
Parent(s):
10dfd97
Updated inference.py
Browse files- apps/inference.py +4 -0
apps/inference.py
CHANGED
@@ -35,6 +35,10 @@ def app():
|
|
35 |
"This demo uses multiple hindi transformer models for Masked Language Modelling (MLM)."
|
36 |
)
|
37 |
|
|
|
|
|
|
|
|
|
38 |
target_text_path = "./mlm_custom/mlm_targeted_text.csv"
|
39 |
target_text_df = pd.read_csv(target_text_path)
|
40 |
|
|
|
35 |
"This demo uses multiple hindi transformer models for Masked Language Modelling (MLM)."
|
36 |
)
|
37 |
|
38 |
+
models_list = list(cfg["models"].keys())
|
39 |
+
|
40 |
+
models = st.multiselect("Choose models", models_list, models_list[0],)
|
41 |
+
|
42 |
target_text_path = "./mlm_custom/mlm_targeted_text.csv"
|
43 |
target_text_df = pd.read_csv(target_text_path)
|
44 |
|