Hugging Face's logo

Spaces:
Rooobert
/
Nvidia_RAG_pdf_R
Sleeping

App Files Community
1
Rooobert commited on
Commit
3945853
·
verified ·
1 Parent(s): 7681671

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
10
 
11
  # Configure Gemini API
12
- genai.configure(api_key=os.getenv("AIzaSyCgfXYWyJjK8YrApUjO6433P3WAdpwQi0Y"))
13
 
14
  # Load Mistral model
15
  model_path = "nvidia/Mistral-NeMo-Minitron-8B-Base"
 
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
10
 
11
  # Configure Gemini API
12
+ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
13
 
14
  # Load Mistral model
15
  model_path = "nvidia/Mistral-NeMo-Minitron-8B-Base"