Update app.py

app.py

@@ -43,10 +43,12 @@ model_2 = AutoModelForSeq2SeqLM.from_pretrained(model_type_2, use_auth_token='ap
 #Defining a function to generate the headlines
 def headline_generator_2(file):
   input_text = clean_text(file.name)
+  input_text = sent_tokenize(input_text)
+  text = ''.join(input_text[:5])
 
   with tokenizer_2.as_target_tokenizer():
         batch = tokenizer_2(
-            input_text, truncation=True, padding="longest", return_tensors="pt"
+            text, truncation=True, padding="longest", return_tensors="pt"
         )
 
   translated = model_2.generate(**batch)