jslin09 commited on
Commit
2ace237
1 Parent(s): 528c89f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -7,7 +7,13 @@ generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-frau
7
  def generate(text):
8
  seed = random.randint(10, 500)
9
  set_seed(seed)
10
- result = generator(text, max_length=500, num_return_sequences=1)
 
 
 
 
 
 
11
  return result[0]["generated_text"]
12
 
13
  examples = [
 
7
  def generate(text):
8
  seed = random.randint(10, 500)
9
  set_seed(seed)
10
+ result = generator(text,
11
+ max_length=500,
12
+ num_return_sequences=1,
13
+ do_sample=True,
14
+ temperature=0.75,
15
+ top_k=50,
16
+ top_p=0.9)
17
  return result[0]["generated_text"]
18
 
19
  examples = [