aleneum commited on
Commit
a37f1f4
β€’
1 Parent(s): d687a03

add output

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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)