Update app.py
Browse files
app.py
CHANGED
@@ -215,12 +215,12 @@ async def upload_pdf(
|
|
215 |
|
216 |
# Générer un embedding pour l'ensemble du texte si le modèle est disponible
|
217 |
embedding = None
|
218 |
-
if
|
219 |
try:
|
220 |
# Limiter la taille du texte si nécessaire
|
221 |
max_length = 5000
|
222 |
truncated_text = text_content[:max_length]
|
223 |
-
embedding =
|
224 |
except Exception as e:
|
225 |
print(f"Erreur lors de la génération de l'embedding: {str(e)}")
|
226 |
|
|
|
215 |
|
216 |
# Générer un embedding pour l'ensemble du texte si le modèle est disponible
|
217 |
embedding = None
|
218 |
+
if embedding_model:
|
219 |
try:
|
220 |
# Limiter la taille du texte si nécessaire
|
221 |
max_length = 5000
|
222 |
truncated_text = text_content[:max_length]
|
223 |
+
embedding = embedding_model.encode(truncated_text).tolist()
|
224 |
except Exception as e:
|
225 |
print(f"Erreur lors de la génération de l'embedding: {str(e)}")
|
226 |
|