Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ import string
|
|
10 |
tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", max_length=128, padding=True, pad_to_max_length = True, truncation=True)
|
11 |
model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
|
12 |
|
13 |
-
|
14 |
def next_word(text, pipe):
|
15 |
res_dict= {
|
16 |
'Word':[],
|
|
|
10 |
tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", max_length=128, padding=True, pad_to_max_length = True, truncation=True)
|
11 |
model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
|
12 |
|
13 |
+
#@st.cache
|
14 |
def next_word(text, pipe):
|
15 |
res_dict= {
|
16 |
'Word':[],
|