Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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,
|