Spaces:

publichealthsurveillance
/

PHS-BERT

Runtime error

publichealthsurveillance commited on May 16, 2022

Commit

3d3d5f3

•

1 Parent(s): 3aaf0d3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,10 @@
 import ktrain
 import gradio as gr
-model = ktrain.load_predictor('bert').model
 def func(text):
-	results = model.predict('James Gandolfini was a great actor.')
 	results = [int(element) for element in results]
 	return results

 import ktrain
 import gradio as gr
+predictor = ktrain.load_predictor('bert')
 def func(text):
+	results = predictor.predict(text)
 	results = [int(element) for element in results]
 	return results