publichealthsurveillance commited on
Commit
d99cb75
1 Parent(s): 23e9c52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,8 @@ import gradio as gr
4
  model = ktrain.load_predictor('bert').model
5
 
6
  def func(text):
7
- return model.predict(text)
 
 
8
 
9
  gr.Interface(fn=func, inputs="text", outputs="text").launch()
 
4
  model = ktrain.load_predictor('bert').model
5
 
6
  def func(text):
7
+ results = model.predict(text)
8
+ results = [int(element) for element in results]
9
+ return results
10
 
11
  gr.Interface(fn=func, inputs="text", outputs="text").launch()