Spaces:
Running
Running
shainaraza
commited on
Commit
•
c156871
1
Parent(s):
8beec99
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def get_debiased_sequence(prompt):
|
|
60 |
do_sample=True,
|
61 |
top_k=10,
|
62 |
num_return_sequences=1,
|
63 |
-
eos_token_id=
|
64 |
max_length=len(prompt.split(" ")) + len(input_text.split(" ")) + 100, # Increased max_length
|
65 |
)
|
66 |
res = sequences[0]['generated_text']
|
|
|
60 |
do_sample=True,
|
61 |
top_k=10,
|
62 |
num_return_sequences=1,
|
63 |
+
eos_token_id=debias_tokenizer.eos_token_id,
|
64 |
max_length=len(prompt.split(" ")) + len(input_text.split(" ")) + 100, # Increased max_length
|
65 |
)
|
66 |
res = sequences[0]['generated_text']
|