hyechanjun commited on
Commit
5805a27
·
1 Parent(s): e5d11d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=2)
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):