Header probs added in BLOCK

This commit is contained in:
Loweel 2019-12-06 16:11:17 +01:00
parent 5cddb90d89
commit b20b50553a

View file

@ -20,6 +20,7 @@ func handler(p *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request)
switch action {
case "BLOCK", "BLOCKLEARN":
p.ModifyResponse = blockAndlearn
w.Header().Set("Probabilities", fmt.Sprintf("%v ", probs))
log.Println("Request Blocked")
p.ServeHTTP(w, r)