acbdkk commited on
Commit
05976e6
1 Parent(s): f9cae74

Update app.py

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