Spaces:
Sleeping
Sleeping
Commit
·
f8732ac
1
Parent(s):
6ee54d9
Embel Model BAAI/bge-small-en-v1.5
Browse files- app.py +7 -0
- 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
|