WhiteAngels
commited on
Commit
•
a647e6a
1
Parent(s):
fe59bf9
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ task = st.sidebar.selectbox("Görev Seç", task_list)
|
|
22 |
|
23 |
# Model listesi ve model seçimi
|
24 |
model_mapping = {
|
25 |
-
'Metin Sınıflandırma': '
|
26 |
'Metin Analizi': 'akdeniz27/convbert-base-turkish-cased-ner',
|
27 |
'Duygu Analizi': 'akdeniz27/xlm-roberta-base-turkish-ner',
|
28 |
'Metin Oluşturma': 'xlm-roberta-large-finetuned-conll03-english'
|
@@ -103,11 +103,10 @@ if Run_Button and input_text != "":
|
|
103 |
|
104 |
for entity in output_comb:
|
105 |
label = entity["entity_group"]
|
106 |
-
# Make sure the label is not empty
|
107 |
if label not in ["PERSON", "NORP", "FAC", "ORG", "GPE", "LOC", "PRODUCT", "EVENT", "WORK_OF_ART", "LAW", "LANGUAGE", "DATE", "TIME", "PERCENT", "MONEY", "QUANTITY", "ORDINAL", "CARDINAL", "MISC"]:
|
108 |
label = "UNKNOWN"
|
109 |
spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": label})
|
110 |
|
111 |
html = displacy.render(spacy_display, style="ent", minify=True, manual=True, options={"ents": ["PERSON", "NORP", "FAC", "ORG", "GPE", "LOC", "PRODUCT", "EVENT", "WORK_OF_ART", "LAW", "LANGUAGE", "DATE", "TIME", "PERCENT", "MONEY", "QUANTITY", "ORDINAL", "CARDINAL", "MISC"]})
|
112 |
style = "<style>mark.entity { display: inline-block }</style>"
|
113 |
-
st.write(f"{style}{get_html(html)}", unsafe_allow_html=True)
|
|
|
22 |
|
23 |
# Model listesi ve model seçimi
|
24 |
model_mapping = {
|
25 |
+
'Metin Sınıflandırma': 'dbmdz/bert-base-turkish-cased',
|
26 |
'Metin Analizi': 'akdeniz27/convbert-base-turkish-cased-ner',
|
27 |
'Duygu Analizi': 'akdeniz27/xlm-roberta-base-turkish-ner',
|
28 |
'Metin Oluşturma': 'xlm-roberta-large-finetuned-conll03-english'
|
|
|
103 |
|
104 |
for entity in output_comb:
|
105 |
label = entity["entity_group"]
|
|
|
106 |
if label not in ["PERSON", "NORP", "FAC", "ORG", "GPE", "LOC", "PRODUCT", "EVENT", "WORK_OF_ART", "LAW", "LANGUAGE", "DATE", "TIME", "PERCENT", "MONEY", "QUANTITY", "ORDINAL", "CARDINAL", "MISC"]:
|
107 |
label = "UNKNOWN"
|
108 |
spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": label})
|
109 |
|
110 |
html = displacy.render(spacy_display, style="ent", minify=True, manual=True, options={"ents": ["PERSON", "NORP", "FAC", "ORG", "GPE", "LOC", "PRODUCT", "EVENT", "WORK_OF_ART", "LAW", "LANGUAGE", "DATE", "TIME", "PERCENT", "MONEY", "QUANTITY", "ORDINAL", "CARDINAL", "MISC"]})
|
111 |
style = "<style>mark.entity { display: inline-block }</style>"
|
112 |
+
st.write(f"{style}{get_html(html)}", unsafe_allow_html=True)
|