JSON Format Fixes

main.py

@@ -37,11 +37,11 @@ def predict(input):
     rets = logits.argmax(dim = 1).tolist()
 
     labels = ["positive", "neutral", "negative"]
-    return {[labels[ret] for ret in rets]}
+    return [labels[ret] for ret in rets]
 
 @app.post("/predict")
 def get_sentiment(data: TextInput):
-    return predict(data.text)
+    return {predict(data.text)}
 
 if __name__ == "__main__":
     import uvicorn