Skip to content

Instantly share code, notes, and snippets.

@willprice76
Last active August 27, 2020 10:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save willprice76/53c6dcccfa12fb105845a76ea8f770af to your computer and use it in GitHub Desktop.
Save willprice76/53c6dcccfa12fb105845a76ea8f770af to your computer and use it in GitHub Desktop.
Adding Spark NLP initialization to controller
package org.example.sparknlp;
import org.apache.spark.sql.SparkSession;
import com.johnsnowlabs.nlp.LightPipeline;
import com.johnsnowlabs.nlp.SparkNLP;
import com.johnsnowlabs.nlp.pretrained.PretrainedPipeline;
import com.sun.org.slf4j.internal.Logger;
import com.sun.org.slf4j.internal.LoggerFactory;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class SparkNlpController {
private static final Logger LOG = LoggerFactory.getLogger(SparkNlpController.class);
private LightPipeline scoringPipeline;
private SparkSession spark;
public SparkNlpController() {
spark = SparkNLP.start(false, false);
scoringPipeline = new PretrainedPipeline("analyze_sentiment", "en").lightModel();
}
@GetMapping("/hello")
public String hello() {
return "Hello world";
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment