Funbi commited on
Commit
9ee0429
1 Parent(s): 4707bee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,9 +20,10 @@ def generate(prompt,textCount=40):
20
  textCount = 40
21
  generator = pipeline('text-generation', model="facebook/opt-1.3b", do_sample=True, num_return_sequences=2, max_length=textCount)
22
  out = generator(prompt)
 
23
 
24
 
25
- return out
26
 
27
 
28
 
 
20
  textCount = 40
21
  generator = pipeline('text-generation', model="facebook/opt-1.3b", do_sample=True, num_return_sequences=2, max_length=textCount)
22
  out = generator(prompt)
23
+ bout = f"{out[0]['generated_text']} \n {out[1]['generated_text']}"
24
 
25
 
26
+ return bout
27
 
28
 
29