amira01 commited on
Commit
6fee022
·
verified ·
1 Parent(s): 0ebb86a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ def load_api_key():
14
  if "OPENROUTER_API_KEY" in os.environ:
15
  return os.getenv("OPENROUTER_API_KEY")
16
  raise ValueError("API key not found in environment variables")
17
-
18
  # Process PDF files
19
  def process_pdfs(files):
20
  all_chunks = []
@@ -52,7 +52,7 @@ def process_pdfs(files):
52
  def load_model():
53
  return ChatOpenAI(
54
  base_url="https://openrouter.ai/api/v1",
55
- api_key=load_api_key(),
56
  model="mistralai/mistral-7b-instruct",
57
  temperature=0.3
58
  )
 
14
  if "OPENROUTER_API_KEY" in os.environ:
15
  return os.getenv("OPENROUTER_API_KEY")
16
  raise ValueError("API key not found in environment variables")
17
+ OPENROUTER_API_KEY=load_api_key()
18
  # Process PDF files
19
  def process_pdfs(files):
20
  all_chunks = []
 
52
  def load_model():
53
  return ChatOpenAI(
54
  base_url="https://openrouter.ai/api/v1",
55
+ api_key=OPENROUTER_API_KEY,
56
  model="mistralai/mistral-7b-instruct",
57
  temperature=0.3
58
  )