Demosthene-OR commited on
Commit
41b5a3d
1 Parent(s): 29b22b6
Files changed (2) hide show
  1. main_dl.py +6 -5
  2. 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
- tiktoken
31
- scikit-learn
 
27
  filesplit==4.0.1
28
  pydot==2.0.0
29
  graphviz==0.20.1
30
+ scikit-learn==1.1.3