lawallanre commited on
Commit
2d5722f
·
verified ·
1 Parent(s): 949a8a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -12,14 +12,14 @@ from typing import List
12
  load_dotenv()
13
 
14
  # Ensure API key is set
15
- OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
16
- if not OPENAI_API_KEY:
17
- raise ValueError("OPENAI_API_KEY is missing. Set it in the .env file.")
18
 
19
- # Initialize OpenAI model
20
- llm = OpenAI(model="gpt-4o")
21
 
22
- # llm = Groq(model="llama3-70b-8192", api_key=os.getenv("GROQ_API_KEY"))
23
 
24
 
25
  # Initialize FastAPI app
 
12
  load_dotenv()
13
 
14
  # Ensure API key is set
15
+ # OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
16
+ # if not OPENAI_API_KEY:
17
+ # raise ValueError("OPENAI_API_KEY is missing. Set it in the .env file.")
18
 
19
+ # # Initialize OpenAI model
20
+ # llm = OpenAI(model="gpt-4o")
21
 
22
+ llm = Groq(model="llama3-70b-8192", api_key=os.getenv("GROQ_API_KEY"))
23
 
24
 
25
  # Initialize FastAPI app