vives commited on
Commit
4979bc7
·
1 Parent(s): b717ac4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
18
  #streamlit stuff
19
  tags = st.text_input("Input tags separated by commas")
20
  text = st.text_input("Input text to classify")
21
- = st.slider("Number of key phrases to extract from text", 10,30,20)
22
  #Methods for tag processing
23
  def pool_embeddings(out, tok):
24
  embeddings = out["hidden_states"][-1]
 
18
  #streamlit stuff
19
  tags = st.text_input("Input tags separated by commas")
20
  text = st.text_input("Input text to classify")
21
+ topkp = st.slider("Number of key phrases to extract from text", 10,30,20)
22
  #Methods for tag processing
23
  def pool_embeddings(out, tok):
24
  embeddings = out["hidden_states"][-1]