Update app.py

app.py

@@ -11,13 +11,13 @@ examples = [
 def ner(text):
     doc = nlp(text)
     final_output = []
-    flagged_categories = ['CARDINAL', 'DATE', 'MONEY', 'PERCENT', 'QUANTITY', 'TIME', 'ORDINAL']
+    flagged_categories = ["CARDINAL", "DATE", "MONEY", "PERCENT", "QUANTITY", "TIME", "ORDINAL"]
     
     for ent in doc.ents:
       label = ent.label_
   
       if label not in flagged_categories:
-          output = {'entity': ent.label_, 'word': ent.text, 'start': int(ent.start_char), 'end': int(ent.end_char)}
+          output = {'entity': ent.label_, "word": ent.text, "start": int(ent.start_char), "end": int(ent.end_char)}
           final_output.append(output)
         
     return {"text": text, "entities": final_output}    
@@ -27,5 +27,6 @@ demo = gr.Interface(ner,
              gr.HighlightedText(),
              examples=examples)
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
   demo.launch(debug=True)