abdullahmubeen10 commited on
Commit
ca3043f
Β·
verified Β·
1 Parent(s): 2e62a3e

Update Demo.py

Browse files
Files changed (1) hide show
  1. Demo.py +1 -1
Demo.py CHANGED
@@ -63,7 +63,7 @@ def create_pipeline(model):
63
  .setOutputCol("ner_chunk")
64
 
65
  nlpPipeline = Pipeline(stages=[documentAssembler, sentenceDetector, tokenizer, tokenClassifier, ner_converter])
66
- return pipeline
67
 
68
  def fit_data(pipeline, data):
69
  empty_df = spark.createDataFrame([['']]).toDF('text')
 
63
  .setOutputCol("ner_chunk")
64
 
65
  nlpPipeline = Pipeline(stages=[documentAssembler, sentenceDetector, tokenizer, tokenClassifier, ner_converter])
66
+ return nlpPipeline
67
 
68
  def fit_data(pipeline, data):
69
  empty_df = spark.createDataFrame([['']]).toDF('text')