Update api.py

api.py

@@ -262,7 +262,10 @@ async def geo_sensor_classify(query: GeoSenQuery):
                              query.pesticide_usage_ml,query.total_days,query.yield_kg_per_hectare,query.latitude,query.longitude,
                              query.NDVI_index])
 
-        return {"answer": np.argmax(logits),"logits":logits}
+        return {
+            "answer": int(np.argmax(logits)),
+            "logits": logits.tolist() if isinstance(logits, np.ndarray) else float(logits)
+        }
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))