Spaces:
Runtime error
Runtime error
prateekagrawal
commited on
Commit
•
10dfd97
1
Parent(s):
02f2597
Updated inference.py
Browse files- apps/inference.py +3 -51
apps/inference.py
CHANGED
@@ -40,35 +40,6 @@ def app():
|
|
40 |
|
41 |
texts = target_text_df["text"]
|
42 |
|
43 |
-
st.markdown("""## Select any of the following text : """)
|
44 |
-
masked_text = st.selectbox("", texts)
|
45 |
-
|
46 |
-
st.write("You selected:", masked_text)
|
47 |
-
|
48 |
-
selected_models = st.multiselect(
|
49 |
-
"Choose models",
|
50 |
-
[
|
51 |
-
"flax-community/roberta-hindi",
|
52 |
-
"mrm8488/HindiBERTa",
|
53 |
-
"neuralspace-reverie/indic-transformers-hi-bert",
|
54 |
-
"surajp/RoBERTa-hindi-guj-san",
|
55 |
-
],
|
56 |
-
["flax-community/roberta-hindi"],
|
57 |
-
)
|
58 |
-
|
59 |
-
models_list = list(cfg["models"].keys())
|
60 |
-
|
61 |
-
models = st.multiselect(
|
62 |
-
"Choose models",
|
63 |
-
models_list,
|
64 |
-
models_list[0],
|
65 |
-
)
|
66 |
-
|
67 |
-
target_text_path = "./mlm_custom/mlm_targeted_text.csv"
|
68 |
-
target_text_df = pd.read_csv(target_text_path)
|
69 |
-
|
70 |
-
texts = target_text_df["text"]
|
71 |
-
|
72 |
st.sidebar.title("Hindi MLM")
|
73 |
|
74 |
pick_random = st.sidebar.checkbox("Pick any random text")
|
@@ -92,7 +63,9 @@ def app():
|
|
92 |
|
93 |
for selected_model in models:
|
94 |
|
95 |
-
filled_sentence, filled_token = load_model(
|
|
|
|
|
96 |
model_names.append(selected_model)
|
97 |
filled_tokens.append(filled_token)
|
98 |
filled_masked_texts.append(filled_sentence)
|
@@ -102,24 +75,3 @@ def app():
|
|
102 |
results_df["Filled Text"] = filled_masked_texts
|
103 |
|
104 |
st.table(results_df)
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
if st.button("Fill the Mask!"):
|
109 |
-
with st.spinner("Filling the Mask..."):
|
110 |
-
models = []
|
111 |
-
predicted_tokens = []
|
112 |
-
predicted_sentence = []
|
113 |
-
for i in range(len(selected_models)):
|
114 |
-
filled_sentence = load_model(masked_text, selected_models[i])
|
115 |
-
# st.write(filled_sentence)
|
116 |
-
models.append(selected_models[i])
|
117 |
-
predicted_tokens.append(filled_sentence[0]["token_str"])
|
118 |
-
predicted_sentence.append(filled_sentence[0]["sequence"])
|
119 |
-
|
120 |
-
results_df = pd.DataFrame()
|
121 |
-
results_df["Model Name"] = models
|
122 |
-
results_df["Predicted Word"] = predicted_tokens
|
123 |
-
results_df["Sentence"] = predicted_sentence
|
124 |
-
st.dataframe(results_df)
|
125 |
-
|
|
|
40 |
|
41 |
texts = target_text_df["text"]
|
42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
43 |
st.sidebar.title("Hindi MLM")
|
44 |
|
45 |
pick_random = st.sidebar.checkbox("Pick any random text")
|
|
|
63 |
|
64 |
for selected_model in models:
|
65 |
|
66 |
+
filled_sentence, filled_token = load_model(
|
67 |
+
masked_text, cfg["models"][selected_model]
|
68 |
+
)
|
69 |
model_names.append(selected_model)
|
70 |
filled_tokens.append(filled_token)
|
71 |
filled_masked_texts.append(filled_sentence)
|
|
|
75 |
results_df["Filled Text"] = filled_masked_texts
|
76 |
|
77 |
st.table(results_df)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|