Doron Adler
commited on
Commit
•
a768034
1
Parent(s):
77d43d7
Minor update (Trace log, remove streamlit version from requirenments file)
Browse files- app.py +1 -3
- 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
|
2 |
transformers
|
3 |
tokenizers
|
4 |
torch
|
|
|
1 |
+
streamlit
|
2 |
transformers
|
3 |
tokenizers
|
4 |
torch
|