WhiteAngels commited on
Commit
a647e6a
1 Parent(s): fe59bf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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': 'akdeniz27/bert-base-turkish-cased-ner',
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)