Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def genQuestion(model_choice, context):
|
|
35 |
# descriptions = "Reverse interview question is a model that..."
|
36 |
|
37 |
inputs = tok(context, return_tensors="pt")
|
38 |
-
output = model.generate(inputs["input_ids"], num_beams=16, max_length=64, min_length=9, num_return_sequences=
|
39 |
final_output = ''
|
40 |
print(output)
|
41 |
for i in range(4):
|
|
|
35 |
# descriptions = "Reverse interview question is a model that..."
|
36 |
|
37 |
inputs = tok(context, return_tensors="pt")
|
38 |
+
output = model.generate(inputs["input_ids"], num_beams=16, max_length=64, min_length=9, num_return_sequences=16, diversity_penalty = 1.0, num_beam_groups=4)
|
39 |
final_output = ''
|
40 |
print(output)
|
41 |
for i in range(4):
|