Spaces:
Sleeping
Sleeping
add output
Browse files
app.py
CHANGED
@@ -114,10 +114,14 @@ if generate:
|
|
114 |
with st.spinner("Generiere Zitat..."):
|
115 |
query = prompt_format.format(
|
116 |
system_prompt=system_prompt,
|
117 |
-
temperature=0.01,
|
118 |
prompt=st.session_state["author"],
|
119 |
)
|
|
|
|
|
120 |
output = st.session_state["model"](
|
121 |
query, stop="<|im_end|>", max_new_tokens=200, **modes[mode]
|
122 |
)
|
|
|
|
|
|
|
123 |
st.success(output)
|
|
|
114 |
with st.spinner("Generiere Zitat..."):
|
115 |
query = prompt_format.format(
|
116 |
system_prompt=system_prompt,
|
|
|
117 |
prompt=st.session_state["author"],
|
118 |
)
|
119 |
+
print("=" * 20)
|
120 |
+
print(query)
|
121 |
output = st.session_state["model"](
|
122 |
query, stop="<|im_end|>", max_new_tokens=200, **modes[mode]
|
123 |
)
|
124 |
+
print("-" * 20)
|
125 |
+
print(output)
|
126 |
+
print("=" * 20)
|
127 |
st.success(output)
|