TymaaHammouda commited on
Commit
0059ef4
·
1 Parent(s): 67bd9a3

Remove evaluate_synonyms

Browse files
Files changed (1) hide show
  1. app.py +17 -17
app.py CHANGED
@@ -29,36 +29,36 @@ download_file_from_hf("SinaLab/ArabGlossBERT","four_grams.pickle")
29
  download_file_from_hf("SinaLab/ArabGlossBERT","five_grams.pickle")
30
 
31
  from sinatools.synonyms.synonyms_generator import extend_synonyms
32
- from sinatools.synonyms.synonyms_generator import evaluate_synonyms
33
  from sinatools.morphology.morph_analyzer import analyze
34
 
35
  class synonymsRequest(BaseModel):
36
  synset: str
37
  useALMA: str
38
  level: str
39
- serviceType: str
40
 
41
  @app.post("/predict")
42
  def predict(request: synonymsRequest):
43
  synset = request.synset
44
  useALMA = request.useALMA
45
  level = int(request.level)
46
- serviceType = request.serviceType
47
 
48
- if serviceType == "evaluate":
49
- list_of_synon_with_fuzzy_value = evaluate_synonyms(synset, level)
50
- content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
51
- else:
52
- final_synset = []
53
- if useALMA == True:
54
- synonyms = synset.split("|")
55
- for synonym in synonyms:
56
- final_synset.append(analyze(synonym.strip(), language = "MSA", task = "lemmatization", flag = "1")[0]["lemma"])
57
- list_of_synon_with_fuzzy_value = extend_synonyms(" | ".join(final_synset), level)
58
- else:
59
- list_of_synon_with_fuzzy_value = extend_synonyms(synset, level)
60
-
61
- content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
62
 
63
  return JSONResponse(
64
  content=content,
 
29
  download_file_from_hf("SinaLab/ArabGlossBERT","five_grams.pickle")
30
 
31
  from sinatools.synonyms.synonyms_generator import extend_synonyms
32
+ # from sinatools.synonyms.synonyms_generator import evaluate_synonyms
33
  from sinatools.morphology.morph_analyzer import analyze
34
 
35
  class synonymsRequest(BaseModel):
36
  synset: str
37
  useALMA: str
38
  level: str
39
+ # serviceType: str
40
 
41
  @app.post("/predict")
42
  def predict(request: synonymsRequest):
43
  synset = request.synset
44
  useALMA = request.useALMA
45
  level = int(request.level)
46
+ # serviceType = request.serviceType
47
 
48
+ # if serviceType == "evaluate":
49
+ # list_of_synon_with_fuzzy_value = evaluate_synonyms(synset, level)
50
+ # content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
51
+ # else:
52
+ final_synset = []
53
+ if useALMA == True:
54
+ synonyms = synset.split("|")
55
+ for synonym in synonyms:
56
+ final_synset.append(analyze(synonym.strip(), language = "MSA", task = "lemmatization", flag = "1")[0]["lemma"])
57
+ list_of_synon_with_fuzzy_value = extend_synonyms(" | ".join(final_synset), level)
58
+ else:
59
+ list_of_synon_with_fuzzy_value = extend_synonyms(synset, level)
60
+
61
+ content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
62
 
63
  return JSONResponse(
64
  content=content,