Namitg02 commited on
Commit
0e0a177
·
verified ·
1 Parent(s): 18221a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -167,11 +167,12 @@ def talk(prompt, history):
167
  # print("check11")
168
  # start a thread
169
  outputs = []
170
- print(model_input)
171
- print(model.tokenize(messages))
172
- tokens = model.tokenize(messages)
173
- for token in model.generate(tokens):
174
- print(model.detokenize([token]))
 
175
  # input_ids = tokenizer(*messages)
176
  # print(model.generate(tensor([[ 1, 529, 29989, 5205, 29989]])))
177
  # start = time.time()
 
167
  # print("check11")
168
  # start a thread
169
  outputs = []
170
+ outputs = model_input
171
+ return outputs
172
+ # print(model.tokenize(messages))
173
+ # tokens = model.tokenize(messages)
174
+ # for token in model.generate(tokens):
175
+ # print(model.detokenize([token]))
176
  # input_ids = tokenizer(*messages)
177
  # print(model.generate(tensor([[ 1, 529, 29989, 5205, 29989]])))
178
  # start = time.time()