Bhanumani12 commited on
Commit
f3f37ce
·
verified ·
1 Parent(s): c98e4d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,12 +2,17 @@ import gradio as gr
2
  from transformers import pipeline
3
  import openai
4
  import os
 
5
 
6
- # Get OpenAI API key securely (without fallback to dummy key)
 
 
 
7
  openai_api_key = os.getenv("OPENAI_API_KEY")
8
  if not openai_api_key:
9
  raise ValueError("❌ OPENAI_API_KEY environment variable not set.")
10
 
 
11
  client = openai.OpenAI(api_key=openai_api_key)
12
 
13
  # Load local model for code classification
 
2
  from transformers import pipeline
3
  import openai
4
  import os
5
+ from dotenv import load_dotenv
6
 
7
+ # Load environment variables from .env file
8
+ load_dotenv()
9
+
10
+ # Get OpenAI API key
11
  openai_api_key = os.getenv("OPENAI_API_KEY")
12
  if not openai_api_key:
13
  raise ValueError("❌ OPENAI_API_KEY environment variable not set.")
14
 
15
+ # Initialize OpenAI client
16
  client = openai.OpenAI(api_key=openai_api_key)
17
 
18
  # Load local model for code classification