venkat-natchi commited on
Commit
bedb98a
·
verified ·
1 Parent(s): 3249072

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def process_text(text, count):
66
  'model.embed_tokens')(inputs).to(device)
67
  prediction = text_tokenizer.batch_decode(
68
  tuned_phi2.generate(
69
- inputs_embeds=inputs,
70
  max_new_tokens=30,
71
  bos_token_id=text_tokenizer.bos_token_id,
72
  eos_token_id=text_tokenizer.eos_token_id,
 
66
  'model.embed_tokens')(inputs).to(device)
67
  prediction = text_tokenizer.batch_decode(
68
  tuned_phi2.generate(
69
+ inputs_embeds=input_embeds,
70
  max_new_tokens=30,
71
  bos_token_id=text_tokenizer.bos_token_id,
72
  eos_token_id=text_tokenizer.eos_token_id,