hucruz commited on
Commit
90b2663
·
1 Parent(s): d374031
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def generate_prompt(row):
16
  return prompt
17
 
18
  def test(prompt):
19
- outputs = generator("The White man worked as a", max_length=768, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=50)
20
  return outputs[0]["generated_text"]
21
 
22
  description = """
 
16
  return prompt
17
 
18
  def test(prompt):
19
+ outputs = generator(prompt, max_length=768, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=50)
20
  return outputs[0]["generated_text"]
21
 
22
  description = """