Update app.py

app.py

@@ -26,23 +26,28 @@ pkgs = freeze.freeze()
 for pkg in pkgs: print(pkg)
 class Query(BaseModel):
     smiles :List[str]
-
+class PredictionResponse(BaseModel):
+    predictions: List[int]
 rob_chem_model = ClassificationModel('roberta', 'BasselAhmed/RobertaChemClinToxTuned',use_cuda=False ,args={'evaluate_each_epoch':True , 'evaluate_during_training_verbose':True, 'seed':4})
 
-@app.post("/ToxicityPrediction")
-def c(query:Query):
+@app.post("/ToxicityPrediction/")
+def c(query:Query)->PredictionResponse: 
     print(query)
     print(type(query))
     query_dict = query.dict()
     try:
         predictions, raw_outputs = rob_chem_model.predict([str(query.smiles)])
         print(predictions)
+        print(type({"predictions": predictions}))
         #json_compatible_item_data = jsonable_encoder(predictions[0])
         #return JSONResponse(content=json_compatible_item_data)
         query_dict.update({"predictions":predictions})
+        print(query_dict)
+        print(type(query_dict))
         #answer = {"prediction":predictions[0]}
         #return {'response':answer}
-        return {"predictions": predictions}
+        
+        return predictions
 
     except Exception as e:
         print("Excepted")