Update app.py

app.py

@@ -1,20 +1,29 @@
 from transformers import pipeline
 import gradio as gr
 from gtts import gTTS
+import openai
 
 # Load the Whisper model for speech-to-text
 pipe = pipeline(model="openai/whisper-small")
 
 # Load the text generation model
-text_pipe = pipeline("text2text-generation", model="google/flan-t5-base")
+# text_pipe = pipeline("text2text-generation", model="google/flan-t5-base")
+
+def generate_gpt_response(text):
+    response = openai.Completion.create(
+        engine="text-davinci-003",  # Use the appropriate GPT-3 engine
+        prompt=text,
+        max_tokens=150
+    )
+    return response.choices[0].text.strip()
 
 def transcribe(audio):
     # Transcribe the audio to text
     text = pipe(audio)["text"]
     
     # Generate a response from the transcribed text
-    lm_response = text_pipe(text)[0]["generated_text"]
-    
+    # lm_response = text_pipe(text)[0]["generated_text"]
+    lm_response = generate_gpt_response(text)
     # Convert the response text to speech
     tts = gTTS(lm_response, lang='ko')