fdehlinger commited on
Commit
f8732ac
·
1 Parent(s): 6ee54d9

Embel Model BAAI/bge-small-en-v1.5

Browse files
Files changed (2) hide show
  1. app.py +7 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,8 +1,15 @@
1
  import gradio as gr
 
 
 
2
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
3
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
4
  from llama_index.core import Settings
5
 
 
 
 
 
6
  def greet(name):
7
  return "Hello " + name + "!!"
8
 
 
1
  import gradio as gr
2
+ import os
3
+ import openai
4
+
5
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
6
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
7
  from llama_index.core import Settings
8
 
9
+ Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
10
+
11
+
12
+ openai.api_key = os.environ['OpenAI_ApiKey']
13
  def greet(name):
14
  return "Hello " + name + "!!"
15
 
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  llama-index==0.12.13
2
- llama-index-embeddings-huggingface==0.5.1
 
 
1
  llama-index==0.12.13
2
+ llama-index-embeddings-huggingface==0.5.1
3
+ openai==1.60.0