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 |
+
|