sandeepmajumdar commited on
Commit
92a38e3
1 Parent(s): ef1acb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  def convert(prompt):
5
  model = pipeline('text-generation', model='bigscience/bloom-1b1')
6
- result = model(prompt, max_length=200)
7
  return result[0]['generated_text']
8
 
9
  with gr.Blocks() as bls:
 
3
 
4
  def convert(prompt):
5
  model = pipeline('text-generation', model='bigscience/bloom-1b1')
6
+ result = model(prompt, max_length=200, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, repetition_penalty = 2.0, diversity_penalty = 0.9)
7
  return result[0]['generated_text']
8
 
9
  with gr.Blocks() as bls: