ZeroGPU
Browse files
demo.py
CHANGED
|
@@ -179,8 +179,8 @@ reid = {
|
|
| 179 |
},
|
| 180 |
"tau": {
|
| 181 |
"A4C": 0.9997,
|
| 182 |
-
"PSAX": 0.
|
| 183 |
-
"PLAX": 0.
|
| 184 |
},
|
| 185 |
}
|
| 186 |
|
|
@@ -403,12 +403,12 @@ def check_privacy(latent_image_numpy, class_selection):
|
|
| 403 |
if corr > tau:
|
| 404 |
return (
|
| 405 |
None,
|
| 406 |
-
f"⚠️ **Warning:** Generated image is too similar to training data. Privacy check failed
|
| 407 |
)
|
| 408 |
else:
|
| 409 |
return (
|
| 410 |
latent_image_numpy,
|
| 411 |
-
f"✅ **Success:** Generated image passed privacy check
|
| 412 |
)
|
| 413 |
|
| 414 |
|
|
|
|
| 179 |
},
|
| 180 |
"tau": {
|
| 181 |
"A4C": 0.9997,
|
| 182 |
+
"PSAX": 0.9997,
|
| 183 |
+
"PLAX": 0.9997,
|
| 184 |
},
|
| 185 |
}
|
| 186 |
|
|
|
|
| 403 |
if corr > tau:
|
| 404 |
return (
|
| 405 |
None,
|
| 406 |
+
f"⚠️ **Warning:** Generated image is too similar to training data. Privacy check failed.",
|
| 407 |
)
|
| 408 |
else:
|
| 409 |
return (
|
| 410 |
latent_image_numpy,
|
| 411 |
+
f"✅ **Success:** Generated image passed privacy check.",
|
| 412 |
)
|
| 413 |
|
| 414 |
|