Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 from llama_index.llms.gemini import Gemini
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
 import os 
-
+from llama_index.embeddings.gemini import GeminiEmbedding
 #os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
 
 st.set_page_config(page_title="Chat with the Streamlit docs, powered by LlamaIndex", page_icon="🦙", layout="centered", initial_sidebar_state="auto", menu_items=None)
@@ -34,6 +34,10 @@ def load_data():
         your answers technical and based on 
         facts – do not hallucinate features.""",
     )
+    Settings.embed_model = GeminiEmbedding(
+    model="models/embedding-001", embed_batch_size=100
+    )
+    
     index = VectorStoreIndex.from_documents(docs)
     return index