Prat0 commited on
Commit
6c99a95
·
verified ·
1 Parent(s): 7d59a7c

Update pages/Report_Writer.py

Browse files
Files changed (1) hide show
  1. pages/Report_Writer.py +3 -2
pages/Report_Writer.py CHANGED
@@ -10,12 +10,13 @@ import google.generativeai as genai
10
  import os
11
  import PyPDF2
12
  import streamlit_analytics2 as streamlit_analytics
13
-
14
 
15
  # Set up Google API key
16
 
17
  # Configure Google Gemini
18
- Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GOOGLE_API_KEY"), model_name="models/embedding-001")
 
19
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
20
  llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.1, model_name="models/gemini-pro")
21
 
 
10
  import os
11
  import PyPDF2
12
  import streamlit_analytics2 as streamlit_analytics
13
+ from llama_index.embeddings.fastembed import FastEmbedEmbedding
14
 
15
  # Set up Google API key
16
 
17
  # Configure Google Gemini
18
+ #Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GOOGLE_API_KEY"), model_name="models/embedding-001")
19
+ Settings.embed_model = FastEmbedEmbedding(model_name="BAAI/bge-small-en-v1.5")
20
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
21
  llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.1, model_name="models/gemini-pro")
22