Akjava commited on
Commit
a53bf4c
Β·
verified Β·
1 Parent(s): 23280b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -61,10 +61,11 @@ def generate_text(messages):
61
  streamer = TextStreamer(tokenizer, skip_prompt=True)
62
  text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer,torch_dtype=dtype,device_map=device ,streamer=streamer) #pipeline has not to(device)
63
  result = text_generator(messages, max_new_tokens=256, do_sample=True, temperature=0.7)
64
- print(result)
65
  generated_output = ""
66
  for token in result:
67
- generated_output += token["generated_token"]
 
68
  yield generated_output
69
 
70
 
 
61
  streamer = TextStreamer(tokenizer, skip_prompt=True)
62
  text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer,torch_dtype=dtype,device_map=device ,streamer=streamer) #pipeline has not to(device)
63
  result = text_generator(messages, max_new_tokens=256, do_sample=True, temperature=0.7)
64
+ print(f"result={result}")
65
  generated_output = ""
66
  for token in result:
67
+ print(f"token={token}")
68
+ generated_output += token["text"]
69
  yield generated_output
70
 
71