Update warm up with the new input layer name

mostafa · mostafa · commit 0ccb8dd98921 · 2024-10-28T11:30:04.000+01:00
diff --git a/training/warm_up.py b/training/warm_up.py
@@ -48,7 +48,7 @@
     predict_request.model_spec.name = "sqli_model"
     predict_request.model_spec.signature_name = "serving_default"
     if sys.argv[3].lower() == "true":
-        predict_request.inputs["keras_tensor"].CopyFrom(
+        predict_request.inputs["keras_tensor_32"].CopyFrom(
             tensor_util.make_tensor_proto(query_vec, float32)
         )
     else:

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@`
`48`	`48`	`predict_request.model_spec.name = "sqli_model"`
`49`	`49`	`predict_request.model_spec.signature_name = "serving_default"`
`50`	`50`	`if sys.argv[3].lower() == "true":`
`51`		`- predict_request.inputs["keras_tensor"].CopyFrom(`
	`51`	`+ predict_request.inputs["keras_tensor_32"].CopyFrom(`
`52`	`52`	`tensor_util.make_tensor_proto(query_vec, float32)`
`53`	`53`	`)`
`54`	`54`	`else:`