Update app.py

app.py

@@ -20,14 +20,13 @@ app.add_middleware(
 
 class Query(BaseModel):
     smiles :List[str]
-#class PredictionResponse(BaseModel):
-    #predictions: List[int]
 class PredictionResponse(BaseModel):
-    predictions: int
+    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)->list[PredictionResponse]:#->PredictionResponse: 
+def c(query:Query)-> PredictionResponse: 
     print(query)
     print(type(query))
     query_dict = query.dict()
@@ -43,8 +42,8 @@ def c(query:Query)->list[PredictionResponse]:#->PredictionResponse:
         #answer = {"prediction":predictions[0]}
         #return {'response':answer}
         
-        #return {"predictions" :list(predictions)}
-        return list(predictions)
+        return {"predictions" :list(predictions)}
+        #return list(predictions)
     except Exception as e:
         print("Excepted")
         raise HTTPException(detail = str(e) , status_code = 500)