abdullahmubeen10 commited on
Commit
a601b98
·
verified ·
1 Parent(s): 1f40f32

Update Demo.py

Browse files
Files changed (1) hide show
  1. Demo.py +2 -2
Demo.py CHANGED
@@ -143,8 +143,8 @@ output = fit_data(pipeline, text_to_analyze)
143
  # Display matched sentence
144
  st.subheader("Processed output:")
145
 
146
- # Define the abbreviation mapping
147
- abbreviation_mapping = dict(zip('RGC', ['LOC', 'PER', 'MISC', 'ORG']))
148
 
149
  # Extract the NER results
150
  results = {
 
143
  # Display matched sentence
144
  st.subheader("Processed output:")
145
 
146
+ # Define the correct abbreviation mapping
147
+ abbreviation_mapping = {'R': 'PER', 'G': 'ORG', 'C': 'LOC', 'SC': 'MISC'}
148
 
149
  # Extract the NER results
150
  results = {