Demosthene-OR
commited on
Commit
•
41b5a3d
1
Parent(s):
29b22b6
...
Browse files- main_dl.py +6 -5
- requirements.txt +1 -2
main_dl.py
CHANGED
@@ -14,7 +14,7 @@ import string
|
|
14 |
import re
|
15 |
import json
|
16 |
import csv
|
17 |
-
import tiktoken
|
18 |
from sklearn.preprocessing import LabelEncoder
|
19 |
from tensorflow import keras
|
20 |
import keras
|
@@ -244,7 +244,7 @@ def load_all_data():
|
|
244 |
rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
|
245 |
|
246 |
# ==== Language identifier ====
|
247 |
-
|
248 |
def encode_text(textes):
|
249 |
global tokenizer
|
250 |
|
@@ -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")
|
@@ -340,7 +340,8 @@ def affiche_modele(lang_tgt:str,
|
|
340 |
|
341 |
# Retourner l'image en tant que réponse HTTP avec le type de contenu approprié
|
342 |
return Response(content=image_data, media_type="image/png")
|
343 |
-
|
344 |
@api.get('/lang_id_dl', name="Id de langue DL")
|
345 |
async def language_id_dl(sentences:List[str] = Query(..., min_length=1)):
|
346 |
-
return lang_id_dl(sentences)
|
|
|
|
14 |
import re
|
15 |
import json
|
16 |
import csv
|
17 |
+
# import tiktoken
|
18 |
from sklearn.preprocessing import LabelEncoder
|
19 |
from tensorflow import keras
|
20 |
import keras
|
|
|
244 |
rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
|
245 |
|
246 |
# ==== Language identifier ====
|
247 |
+
'''
|
248 |
def encode_text(textes):
|
249 |
global tokenizer
|
250 |
|
|
|
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")
|
|
|
340 |
|
341 |
# Retourner l'image en tant que réponse HTTP avec le type de contenu approprié
|
342 |
return Response(content=image_data, media_type="image/png")
|
343 |
+
'''
|
344 |
@api.get('/lang_id_dl', name="Id de langue DL")
|
345 |
async def language_id_dl(sentences:List[str] = Query(..., min_length=1)):
|
346 |
+
return lang_id_dl(sentences)
|
347 |
+
'''
|
requirements.txt
CHANGED
@@ -27,5 +27,4 @@ sentencepiece==0.1.99
|
|
27 |
filesplit==4.0.1
|
28 |
pydot==2.0.0
|
29 |
graphviz==0.20.1
|
30 |
-
|
31 |
-
scikit-learn
|
|
|
27 |
filesplit==4.0.1
|
28 |
pydot==2.0.0
|
29 |
graphviz==0.20.1
|
30 |
+
scikit-learn==1.1.3
|
|