Komal-patra commited on
Commit
a216bc8
·
verified ·
1 Parent(s): 72fd561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -34,10 +34,10 @@ def generate_text(prompt, max_length=200):
34
  outputs = model.generate(
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
- num_beams=7,
38
- temperature=0.7
39
-
40
  )
 
41
  print(outputs)
42
  # Decode the generated tokens
43
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
34
  outputs = model.generate(
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
+ num_beams=5,
38
+ num_return_sequences=3
 
39
  )
40
+
41
  print(outputs)
42
  # Decode the generated tokens
43
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)