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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ pipeline = pipeline(model="jeevana/EmailSubjectLineGeneration", max_new_tokens=1
8
  def predict(input):
9
  print("pipeline object", pipeline)
10
  prediction = pipeline(input)
11
- print("type of response:", type(input))
 
12
  return prediction
13
 
14
 
 
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