kingarnica commited on
Commit
59c6e91
·
verified ·
1 Parent(s): ff865cb

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -32,9 +32,8 @@ def load_pipeline():
32
 
33
  if user_input:
34
  pipe = load_pipeline()
35
- prompt = f"{system_prompt}
36
- User: {user_input}
37
- Assistant:"
38
  response = pipe(prompt, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
39
  st.markdown("**Response:**")
40
  st.write(response.replace(prompt, ""))
 
 
32
 
33
  if user_input:
34
  pipe = load_pipeline()
35
+ prompt = f"{system_prompt}\nUser: {user_input}\nAssistant:"
 
 
36
  response = pipe(prompt, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
37
  st.markdown("**Response:**")
38
  st.write(response.replace(prompt, ""))
39
+