Update app.py

app.py

@@ -27,9 +27,11 @@ async def c(query:Query):
     try:
         predictions, raw_outputs = rob_chem_model.predict([str(query.query)])
         print(predictions)
-
-        return {"prediction":predictions}
+        answer = {"prediction":predictions[0]}
+        return answer
+        
 
     except Exception as e:
+        print("Excepted")
         raise HTTPException(detail = str(e) , status_code = 500)