jeevana commited on
Commit
1e46059
·
verified ·
1 Parent(s): 7e57d2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ def predict(input):
9
  print("pipeline object", pipeline)
10
  prediction = pipeline(input)
11
  prediction = prediction[0].get("generated_text")
12
- prediction = prediction[len(input)+8]
 
13
  return prediction
14
 
15
 
 
9
  print("pipeline object", pipeline)
10
  prediction = pipeline(input)
11
  prediction = prediction[0].get("generated_text")
12
+ print("1:::", prediction)
13
+ prediction = prediction[len(input)+len('@subject'):]
14
  return prediction
15
 
16