Update app.py

app.py

@@ -1,10 +1,16 @@
+from chromadb.utils import embedding_functions
+import chromadb
 from openai import OpenAI
 import gradio as gr
 import time
 
 anyscale_base_url = "https://api.endpoints.anyscale.com/v1"
+multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="jost/multilingual-e5-base-politics-de")
 
 def predict(api_key, user_input):
+    client = chromadb.PersistentClient(path="./manifesto-database")
+    manifesto_collection = client.get_or_create_collection(name="manifesto-database", embedding_function=multilingual_embeddings)
+    
     prompt = f"""[INST] {user_input} [/INST]"""
     
     client = OpenAI(base_url=anyscale_base_url, api_key=api_key)