test-gradio / app.py
fdehlinger's picture
Embel Model BAAI/bge-small-en-v1.5
f8732ac
raw
history blame
475 Bytes
import gradio as gr
import os
import openai
from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.core import Settings
Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
openai.api_key = os.environ['OpenAI_ApiKey']
def greet(name):
return "Hello " + name + "!!"
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()