Hugging Face's logo

Spaces:
Wootang01
/
Punctuation_capitalization_corrector
Runtime error

App Files Community
Punctuation_capitalization_corrector
File size: 2,120 Bytes 
8fb2743
 
 
 
 
 
c3787f9
 
 
 
 
 
 
 
 
 
 
 
 
8fb2743
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
aeb8864
8fb2743
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
import streamlit as st
from multiprocessing import Process
import json
import requests
import time
import os

def load_models():
  if not is_port_in_use(8080):
    with st.spinner(text="The model is loading."):
      proc = Process(target=start_server, args=(), daemon=True)
      proc.start()
      while not is_port_in_use(8080):
        time.sleep(1)
      st.success("Model server started.")
  else:
    st.success("The model has loaded.")
  st.session_state['models_loaded'] = True

def get_correction(input_text):
  st.markdown(f'##### Corrected text:')
  st.write('')
  correct_request = "http://0.0.0.0:8080/restore?input_sentence="+input_text
  with st.spinner('Wait for it...'):
    correct_response = requests.get(correct_request)
    correct_json = json.loads(correct_response.text)
    corrected_sentence = correct_json["corrected_sentence"]
    result = diff_strings(corrected_sentence,input_text)
  st.markdown(result, unsafe_allow_html=True)
  
def diff_strings(output_text, input_text):
  c_text = ""
  for x in output_text.split(" "):
    if x in input_text.split(" "):
      c_text = c_text + x + " "
    else:
      c_text = c_text + '<span style="font-weight:bold; color:rgb(142, 208, 129);">' + x + '</span>' + " "
    return c_text

if __name__ == "__main__":

  st.title('Punctuation and Capitalization Corrector -- BERT')
  st.markdown("Choose an example or input your own text. The machine will attempt to correct the text's punctuation and capitalization.")
  examples = [
              "my name is clara and i live in berkeley california",
              "in 2018 cornell researchers built a high-powered detector",
              "lorem ipsum has been the industrys standard dummy text ever since the 1500s when an unknown printer took a galley of type and scrambled it to make a type specimen book"
 	          ]
  if not st.session_state['models_loaded']:
    load_models()

  input_text = st.selectbox(
  label="Examples",
  options=examples
  )
  input_text = st.text_input(
    label="Write or paste text",
    value=input_text
  )
  if input_text.strip(): 
    get_correction(input_text)