Hamda commited on
Commit
fa1c92c
1 Parent(s): cd7dcf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- @st.cache
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("Use our model to expand your query based on the DB content")
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)