Update app.py

app.py

@@ -1,23 +1,28 @@
+import openai
 import gradio as gr
 import os
 import subprocess
 from gtts import gTTS
 from pydub import AudioSegment
-from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
 from PIL import Image
 
-generator = pipeline('text-generation', model="checkpoints")
-tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
+openai.api_key = "openai_key"  # Reemplaza "openai_key" con el nombre de tu clave de API de OpenAI
+
 os.environ["TOKENIZERS_PARALLELISM"] = "true"
 
 def generate_output(name, date_of_birth, progress=gr.Progress()):
-    prompt = f"Bienvenido {name}:"
-    input_tokens = tokenizer.encode(prompt, add_special_tokens=True)
-    input_text = tokenizer.decode(input_tokens)
-    gpt2_output = generator(input_text, max_length=60, do_sample=True, temperature=0.6)
-    generated_text = gpt2_output[0]['generated_text']
-    generated_text = generated_text.replace(input_text, "").strip()
-    if len(gpt2_output) == 0 or 'generated_text' not in gpt2_output[0]:
+    prompt = f"{name}, tu mi oráculo astrológico de hoy es:"
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=60,
+        temperature=0.6,
+        n=1,
+        stop=None,
+    )
+    gpt3_output = response.choices[0].text.strip()
+    generated_text = gpt3_output.replace(prompt, "").strip()
+    if len(response.choices) == 0 or 'text' not in response.choices[0]:
         return None, "No se pudo generar el texto."
 
     def _progress(generated, to_generate):