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 |
|