MienOlle commited on
Commit
06912ed
·
1 Parent(s): 81adbfa

More JSON fixes

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -41,7 +41,7 @@ def predict(input):
41
 
42
  @app.post("/predict")
43
  def get_sentiment(data: TextInput):
44
- return {predict(data.text)}
45
 
46
  if __name__ == "__main__":
47
  import uvicorn
 
41
 
42
  @app.post("/predict")
43
  def get_sentiment(data: TextInput):
44
+ return {"predictions": predict(data.text)}
45
 
46
  if __name__ == "__main__":
47
  import uvicorn