Spaces:

valurank
/

Headline_generator

Runtime error

abdulmatinomotoso commited on Jun 10, 2022

Commit

34cdf1f

1 Parent(s): 67ea98b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def headline_generator_2(file):
   with tokenizer_2.as_target_tokenizer():
         batch = tokenizer_2(
-            input_text[:200], truncation=True, padding="longest", return_tensors="pt"
         )
   translated = model_2.generate(**batch)

   with tokenizer_2.as_target_tokenizer():
         batch = tokenizer_2(
+            input_text, truncation=True, padding="longest", return_tensors="pt"
         )
   translated = model_2.generate(**batch)