Update app.py

app.py

@@ -7,10 +7,10 @@ pipeline = pipeline(model="jeevana/EmailSubjectLineGeneration", max_new_tokens=1
 
 def predict(input):
     print("pipeline object", pipeline)
-    prediction = pipeline(input)
+    prediction = pipeline(input+'\n@subject\n')
     prediction = prediction[0].get("generated_text")
     print("1:::", prediction)
-    prediction = prediction[len(input)+len('@subject'):]
+    prediction = prediction[len(input)+len('\n@subject\n'):]
     return prediction