Update app.py

app.py

@@ -1,16 +1,7 @@
-from transformers import pipeline
-from transformers import AutoTokenizer
+import openai
 import tweepy
 import os
-
-
-# Cargar el tokenizador
-tokenizer = AutoTokenizer.from_pretrained("gpt2")
-
-# Tokenizar con truncamiento explícito
-inputs = tokenizer("Tu texto largo aquí", max_length=50, truncation=True, padding=True)
-
-print(inputs)
+from transformers import AutoTokenizer, pipeline
 
 # Autenticación de Twitter
 def authenticate_twitter():
@@ -23,14 +14,21 @@ def authenticate_twitter():
     api = tweepy.API(auth)
     return api
 
-# Función para generar un tweet utilizando Hugging Face GPT-2
+# Cargar y preparar el tokenizador
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+# Asignar el token de padding al token de fin de secuencia
+tokenizer.pad_token = tokenizer.eos_token
+
+# Función para generar un tweet usando GPT-2
 def generate_tweet():
     try:
-        # Usamos Hugging Face para cargar el modelo GPT-2
-        generator = pipeline('text-generation', model='gpt2')
-        prompt = "Create a tweet in the style of a humorous AI agent, keeping it casual and engaging."
-        result = generator(prompt, max_length=50, num_return_sequences=1)
-        tweet = result[0]['generated_text'].strip()
+        # Tokenización con truncamiento y padding
+        inputs = tokenizer("Genera un tweet divertido sobre IA", max_length=50, truncation=True, padding=True)
+
+        # Generación de texto
+        generator = pipeline("text-generation", model="gpt2")
+        tweet = generator(inputs['input_ids'], max_length=50, num_return_sequences=1)[0]['generated_text']
+
         return tweet
 
     except Exception as e:
@@ -47,16 +45,14 @@ def post_tweet(tweet, api):
 
 # Función principal para ejecutar el script
 def main():
-    api_twitter = authenticate_twitter()
-
-    tweet = generate_tweet()  # Genera un nuevo tweet
+    api_twitter = authenticate_twitter()  # Autenticación con Twitter
+    tweet = generate_tweet()  # Generar el tweet
 
     if tweet:  # Verifica si el tweet fue generado correctamente
-        post_tweet(tweet, api_twitter)  # Publica el tweet
+        post_tweet(tweet, api_twitter)  # Publicar el tweet
     else:
         print("No se pudo generar un tweet.")
 
 # Iniciar el programa
 if __name__ == "__main__":
     main()
-