Spaces:
Runtime error
Runtime error
Commit
·
cb0baeb
1
Parent(s):
17f4cff
Update main_dl.py
Browse files- 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)
|
|