Spaces:
Runtime error
Runtime error
Commit
·
5805a27
1
Parent(s):
e5d11d3
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def genQuestion(model_choice, context):
|
|
33 |
# descriptions = "Reverse interview question is a model that..."
|
34 |
|
35 |
inputs = tok(context, return_tensors="pt")
|
36 |
-
output = model.generate(inputs["input_ids"], num_beams=4, max_length=64, min_length=9, num_return_sequences=4, diversity_penalty =1.0, num_beam_groups=
|
37 |
final_output = ''
|
38 |
|
39 |
for i in range(4):
|
|
|
33 |
# descriptions = "Reverse interview question is a model that..."
|
34 |
|
35 |
inputs = tok(context, return_tensors="pt")
|
36 |
+
output = model.generate(inputs["input_ids"], num_beams=4, max_length=64, min_length=9, num_return_sequences=4, diversity_penalty =1.0, num_beam_groups=4)
|
37 |
final_output = ''
|
38 |
|
39 |
for i in range(4):
|