Spaces:
Running
Running
Commit
·
0059ef4
1
Parent(s):
67bd9a3
Remove evaluate_synonyms
Browse files
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 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 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,
|