Demosthene-OR commited on
Commit
cb0baeb
·
1 Parent(s): 17f4cff

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +3 -4
main_dl.py CHANGED
@@ -273,7 +273,7 @@ def init_dl_identifier():
273
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
274
  dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5")
275
  return
276
- '''
277
  def lang_id_dl(sentences):
278
  global dl_model, label_encoder, lan_to_language
279
 
@@ -284,7 +284,7 @@ def lang_id_dl(sentences):
284
  predicted_languages = label_encoder.classes_[predicted_labels_encoded]
285
  if "str" in str(type(sentences)): return lan_to_language[predicted_languages[0]]
286
  else: return [l for l in predicted_languages]
287
- '''
288
  # ==== Endpoints ====
289
 
290
  @api.get('/', name="Vérification que l'API fonctionne")
@@ -339,8 +339,7 @@ def affiche_modele(lang_tgt:str,
339
 
340
  # Retourner l'image en tant que réponse HTTP avec le type de contenu approprié
341
  return Response(content=image_data, media_type="image/png")
342
- '''
343
  @api.get('/lang_id_dl', name="Id de langue DL")
344
  async def language_id_dl(sentences:List[str] = Query(..., min_length=1)):
345
  return lang_id_dl(sentences)
346
- '''
 
273
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
274
  dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5")
275
  return
276
+
277
  def lang_id_dl(sentences):
278
  global dl_model, label_encoder, lan_to_language
279
 
 
284
  predicted_languages = label_encoder.classes_[predicted_labels_encoded]
285
  if "str" in str(type(sentences)): return lan_to_language[predicted_languages[0]]
286
  else: return [l for l in predicted_languages]
287
+
288
  # ==== Endpoints ====
289
 
290
  @api.get('/', name="Vérification que l'API fonctionne")
 
339
 
340
  # Retourner l'image en tant que réponse HTTP avec le type de contenu approprié
341
  return Response(content=image_data, media_type="image/png")
342
+
343
  @api.get('/lang_id_dl', name="Id de langue DL")
344
  async def language_id_dl(sentences:List[str] = Query(..., min_length=1)):
345
  return lang_id_dl(sentences)