Update app.py

app.py

@@ -66,12 +66,16 @@ def load_models():
             model="Helsinki-NLP/opus-mt-tc-big-en-pt",
             device=device
         )
+
+        tokenizer_gpt2 = AutoTokenizer.from_pretrained("gpt2")
+        model_gpt2 = AutoModelForCasualLM.from_pretrained("gpt2")
+        model_gpt2.config.pad_token_id = model_gpt2.config.eos_token_id
         
         models['text_generation'] = pipeline(
             "text-generation",
-            model="gpt2",
-            device=device,
-            max_length=100
+            model=model_gpt2,
+            tokenizer=tokenizer_gpt2,
+            device=device
         )
         
         models['ner'] = pipeline(