MahmoudH commited on
Commit
41e454e
·
1 Parent(s): 8ac5c2a

Add prefix

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ tokenizer = AutoTokenizer.from_pretrained("MahmoudH/t5-v1_1-base-finetuned-sci_s
5
  model = TFAutoModelForSeq2SeqLM.from_pretrained("MahmoudH/t5-v1_1-base-finetuned-sci_summ")
6
 
7
  def predict(text):
 
8
  tokenized_inputs = tokenizer([text])
9
  output = model.generate(
10
  input_ids=tokenized_inputs["input_ids"],
 
5
  model = TFAutoModelForSeq2SeqLM.from_pretrained("MahmoudH/t5-v1_1-base-finetuned-sci_summ")
6
 
7
  def predict(text):
8
+ text = "summarize: " + text
9
  tokenized_inputs = tokenizer([text])
10
  output = model.generate(
11
  input_ids=tokenized_inputs["input_ids"],