Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import pandas as pd
|
|
9 |
tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
|
10 |
model = AutoModelForMaskedLM.from_pretrained("moussaKam/AraBART")
|
11 |
|
12 |
-
|
13 |
def next_word(text, pipe):
|
14 |
res_dict= {
|
15 |
'Word':[],
|
@@ -22,7 +22,7 @@ def next_word(text, pipe):
|
|
22 |
return res_dict
|
23 |
|
24 |
st.title("Predict Next Word")
|
25 |
-
st.write("
|
26 |
default_value = "التاريخ هو تحليل و"
|
27 |
# sent is the the variable holding the user's input
|
28 |
sent = st.text_area("Text", default_value, height = 30)
|
|
|
9 |
tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
|
10 |
model = AutoModelForMaskedLM.from_pretrained("moussaKam/AraBART")
|
11 |
|
12 |
+
#@st.cache
|
13 |
def next_word(text, pipe):
|
14 |
res_dict= {
|
15 |
'Word':[],
|
|
|
22 |
return res_dict
|
23 |
|
24 |
st.title("Predict Next Word")
|
25 |
+
st.write("Expand your query by leveraging various models")
|
26 |
default_value = "التاريخ هو تحليل و"
|
27 |
# sent is the the variable holding the user's input
|
28 |
sent = st.text_area("Text", default_value, height = 30)
|