Prat0 commited on
Commit
f6b3c84
·
verified ·
1 Parent(s): 91b0ff8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ from llama_index.llms.gemini import Gemini
8
  from llama_index.embeddings.gemini import GeminiEmbedding
9
  from llama_index.core.memory import ChatMemoryBuffer
10
  import gradio as gr
 
11
 
12
  def embed_setup():
13
  Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GEMINI_API_KEY"), model_name="models/embedding-001")
 
8
  from llama_index.embeddings.gemini import GeminiEmbedding
9
  from llama_index.core.memory import ChatMemoryBuffer
10
  import gradio as gr
11
+ import os
12
 
13
  def embed_setup():
14
  Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GEMINI_API_KEY"), model_name="models/embedding-001")