Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -41,7 +41,7 @@ def read_root(data: req):
|
|
41 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
42 |
|
43 |
# Generate output using the model
|
44 |
-
output_ids = model.generate(input_ids, max_length=length, num_beams=5, no_repeat_ngram_size=2)
|
45 |
generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
46 |
|
47 |
answer_data = { "answer": generated_text }
|
|
|
41 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
42 |
|
43 |
# Generate output using the model
|
44 |
+
output_ids = model.generate(input_ids, max_length=data.length, num_beams=5, no_repeat_ngram_size=2)
|
45 |
generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
46 |
|
47 |
answer_data = { "answer": generated_text }
|