AxL95 commited on
Commit
ed05016
·
verified ·
1 Parent(s): 3deba5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 embedder:
219
  try:
220
  # Limiter la taille du texte si nécessaire
221
  max_length = 5000
222
  truncated_text = text_content[:max_length]
223
- embedding = embedder.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
 
 
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