Doron Adler commited on
Commit
a768034
1 Parent(s): 77d43d7

Minor update (Trace log, remove streamlit version from requirenments file)

Browse files
Files changed (2) hide show
  1. app.py +1 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -94,8 +94,6 @@ if __name__ == "__main__":
94
 
95
  model.to(device)
96
 
97
- print(f"device:{device}, n_gpu:{n_gpu}, random_seed:{random_seed}")
98
-
99
  st.sidebar.subheader("Configurable parameters")
100
 
101
  max_len = st.sidebar.slider("Max-Length", 0, 512, 256,help="The maximum length of the sequence to be generated.")
@@ -112,7 +110,7 @@ if __name__ == "__main__":
112
  if st.button("Run"):
113
  with st.spinner(text="Generating results..."):
114
  st.subheader("Result")
115
- print(f"maxlen:{max_len}, top_k:{top_k}, top_p:{top_p}")
116
  result = extend(input_text=text,
117
  max_size=int(max_len),
118
  top_k=int(top_k),
 
94
 
95
  model.to(device)
96
 
 
 
97
  st.sidebar.subheader("Configurable parameters")
98
 
99
  max_len = st.sidebar.slider("Max-Length", 0, 512, 256,help="The maximum length of the sequence to be generated.")
 
110
  if st.button("Run"):
111
  with st.spinner(text="Generating results..."):
112
  st.subheader("Result")
113
+ print(f"device:{device}, n_gpu:{n_gpu}, random_seed:{random_seed}, maxlen:{max_len}, top_k:{top_k}, top_p:{top_p}")
114
  result = extend(input_text=text,
115
  max_size=int(max_len),
116
  top_k=int(top_k),
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- streamlit==0.80.0
2
  transformers
3
  tokenizers
4
  torch
 
1
+ streamlit
2
  transformers
3
  tokenizers
4
  torch