Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,6 +13,7 @@ message = st.chat_input('Message')
|
|
13 |
if message:
|
14 |
prompt = f"### User: {message}\n\n### Assistant:\n"
|
15 |
conv+=prompt
|
|
|
16 |
# Run the model as if it were on your computer
|
17 |
inputs = tokenizer(conv, return_tensors="pt")["input_ids"]
|
18 |
outputs = model.generate(inputs, max_new_tokens=256)
|
|
|
13 |
if message:
|
14 |
prompt = f"### User: {message}\n\n### Assistant:\n"
|
15 |
conv+=prompt
|
16 |
+
print(conv)
|
17 |
# Run the model as if it were on your computer
|
18 |
inputs = tokenizer(conv, return_tensors="pt")["input_ids"]
|
19 |
outputs = model.generate(inputs, max_new_tokens=256)
|