Update app.py

app.py

@@ -35,17 +35,17 @@ from transformers import (
 )
 
 #initializing the tokenizer and the model
-tokenizer = AutoTokenizer.from_pretrained("valurank/headline_generator_baseline")
-model = AutoModelForSeq2SeqLM.from_pretrained("valurank/headline_generator_baseline")
+tokenizer = AutoTokenizer.from_pretrained("valurank/final_headline_generator")
+model = AutoModelForSeq2SeqLM.from_pretrained("valurank/final_headline_generator")
 
 #Defining a function to generate the headlines
 def headline_generator_2(file):
-  input_text = clean_text(file)
-  input_text = sent_tokenize(input_text)
-  text = ''.join(input_text[:6])
+  input_text = file
+  #input_text = sent_tokenize(input_text)
+  #text = ''.join(input_text[:6])
   
-  inputs = tokenizer(text,truncation=True, return_tensors="pt")
-  summary_ids = model.generate(inputs["input_ids"])
+  inputs = tokenizer(input_text,truncation=True, return_tensors="pt")
+  summary_ids = model.generate(inputs["input_ids"],min_length=20, max_length=40)
   summary = tokenizer.batch_decode(summary_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
   
   return summary