tomrb commited on
Commit
4f34192
·
1 Parent(s): 2085d11

Update app.py

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