Spaces:
Running
Running
update app.py
Browse files
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 |
+
|