Marvin M. Agüero-Torales commited on
Commit
18996d6
·
1 Parent(s): fa4b438

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ HISTORY_WEIGHT = 100 # set history weight (if found any keyword from history, it
9
 
10
  @st.cache(allow_output_mutation=True, suppress_st_warning=True)
11
  def get_model(model):
12
- return pipeline("fill-mask", model=model, top_k=10)#set the maximum of tokens to be retrieved after each inference to model
13
 
14
  def hash_func(inp):
15
  return True
 
9
 
10
  @st.cache(allow_output_mutation=True, suppress_st_warning=True)
11
  def get_model(model):
12
+ return pipeline("fill-mask", model=model, top_k=10)#s5t the maximum of tokens to be retrieved after each inference to model
13
 
14
  def hash_func(inp):
15
  return True