sandeepmajumdar commited on
Commit
0a26a6b
1 Parent(s): 88c036d

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, repetition_penalty = 2.0) #, max_length=100, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, diversity_penalty = 0.9)
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, repetition_penalty = 2.0, max_length=200, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, diversity_penalty = 0.9)
7
  return result[0]['generated_text']
8
 
9
  with gr.Blocks() as bls: