xavierbarbier commited on
Commit
49db238
·
verified ·
1 Parent(s): 575a790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -5,6 +5,17 @@ import numpy as np
5
  from dotenv import load_dotenv
6
  import os
7
  from litellm import completion
 
 
 
 
 
 
 
 
 
 
 
8
 
9
  # Charger les variables d'environnement depuis un fichier .env (recommandé pour les clés API)
10
  load_dotenv()
 
5
  from dotenv import load_dotenv
6
  import os
7
  from litellm import completion
8
+ from langchain_community.embeddings import HuggingFaceEmbeddings
9
+
10
+ model_name = "bge-small-en-v1.5"
11
+ model_kwargs = {'device': 'cpu'}
12
+ encode_kwargs = {'normalize_embeddings': False}
13
+ embeddings = HuggingFaceEmbeddings(
14
+ #model_name=model_name,
15
+ model_kwargs=model_kwargs,
16
+ encode_kwargs=encode_kwargs
17
+ )
18
+
19
 
20
  # Charger les variables d'environnement depuis un fichier .env (recommandé pour les clés API)
21
  load_dotenv()