Spaces:
Sleeping
Sleeping
Update Demo.py
Browse files
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
|
| 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')
|