Балаганский Никита Николаевич
commited on
Commit
•
9f69cab
1
Parent(s):
9320186
fix
Browse files
app.py
CHANGED
@@ -180,6 +180,8 @@ def main():
|
|
180 |
auth_token = os.environ.get('TOKEN') or True
|
181 |
fp16 = st.checkbox("FP16", value=True)
|
182 |
st.session_state["generated_text"] = None
|
|
|
|
|
183 |
|
184 |
def generate():
|
185 |
text = inference(
|
@@ -192,8 +194,7 @@ def main():
|
|
192 |
fp16=fp16,
|
193 |
act_type=act_type
|
194 |
)
|
195 |
-
|
196 |
-
st.write(st.session_state["generated_text"])
|
197 |
st.button("Generate new", on_click=generate())
|
198 |
|
199 |
|
|
|
180 |
auth_token = os.environ.get('TOKEN') or True
|
181 |
fp16 = st.checkbox("FP16", value=True)
|
182 |
st.session_state["generated_text"] = None
|
183 |
+
st.subheader("Generated text:")
|
184 |
+
st.write(st.session_state["generated_text"])
|
185 |
|
186 |
def generate():
|
187 |
text = inference(
|
|
|
194 |
fp16=fp16,
|
195 |
act_type=act_type
|
196 |
)
|
197 |
+
|
|
|
198 |
st.button("Generate new", on_click=generate())
|
199 |
|
200 |
|