Header probs added in BLOCK
This commit is contained in:
parent
5cddb90d89
commit
b20b50553a
1 changed files with 1 additions and 0 deletions
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in a new issue