Update inference_server.py

inference_server.py

@@ -23,7 +23,7 @@ def predict(input_text: str):
     # Make a prediction on the new input string
     prediction = model.predict([input_text])[0]
 
-    return {"prediction": prediction}
+    return {"prediction": prediction, "num_strings": len(strings)}
     
 # Here you can do things such as load your models