add prompt

aiSum.py

@@ -18,7 +18,7 @@ class Sum():
     def summarize(text_to_summarize):
         # Tokenize our text
         # If you want to run the code in Tensorflow, please remember to return the particular tensors as simply as using return_tensors = 'tf'
-        input_ids = tokenizer(text_to_summarize, return_tensors="pt").input_ids
+        input_ids = tokenizer(f"Summarize: {text_to_summarize}", return_tensors="pt").input_ids
 
         # Generate the output (Here, we use beam search but you can also use any other strategy you like)
         output = model.generate(