tomrb commited on
Commit
9f65801
1 Parent(s): b5fae80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate(text):
27
 
28
  preprocessed_text = preprocess(text)
29
  result = generator(preprocessed_text, max_length=128)
30
- output = re.split(r'\nQuestion:|Answer #1:|Title:',result[0]['generated_text'])[2]
31
 
32
  return output
33
 
 
27
 
28
  preprocessed_text = preprocess(text)
29
  result = generator(preprocessed_text, max_length=128)
30
+ output = re.split(r'\nQuestion:|Answer #1:|Answer #|Title:',result[0]['generated_text'])[2]
31
 
32
  return output
33