shiue2000 commited on
Commit
c3b887e
·
verified ·
1 Parent(s): c1ae8ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  import whisper
3
- import openai
4
  import os
 
5
 
6
  # 🔑 Load OpenAI API key from Hugging Face Space secrets
7
- openai.api_key = os.environ.get("OPENAI_API_KEY")
8
 
9
  # 🎧 Load Whisper model once (small = faster; use "base" for lighter)
10
  whisper_model = whisper.load_model("small")
@@ -27,12 +27,12 @@ def transcribe_and_translate(audio_path):
27
  ]
28
 
29
  try:
30
- response = openai.ChatCompletion.create(
31
- model="gpt-4o-mini", # can change to "gpt-4o" for higher quality
32
  messages=messages,
33
  temperature=0
34
  )
35
- english_text = response["choices"][0]["message"]["content"].strip()
36
  except Exception as e:
37
  english_text = f"⚠️ Translation failed: {e}"
38
 
 
1
  import gradio as gr
2
  import whisper
 
3
  import os
4
+ from openai import OpenAI
5
 
6
  # 🔑 Load OpenAI API key from Hugging Face Space secrets
7
+ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
8
 
9
  # 🎧 Load Whisper model once (small = faster; use "base" for lighter)
10
  whisper_model = whisper.load_model("small")
 
27
  ]
28
 
29
  try:
30
+ response = client.chat.completions.create(
31
+ model="gpt-4o-mini", # can use "gpt-4o" for higher quality
32
  messages=messages,
33
  temperature=0
34
  )
35
+ english_text = response.choices[0].message.content.strip()
36
  except Exception as e:
37
  english_text = f"⚠️ Translation failed: {e}"
38