Spaces:
Running
Running
added softmax temperature for better uncertainty calibration
Browse files- app/main.py +2 -2
- app/static/index.html +3 -3
app/main.py
CHANGED
|
@@ -9,8 +9,8 @@ from PIL import Image
|
|
| 9 |
from .model import load_detector, predict_image
|
| 10 |
from .video import sample_frames
|
| 11 |
|
| 12 |
-
MAX_IMAGE_SIZE_MB =
|
| 13 |
-
MAX_VIDEO_SIZE_MB =
|
| 14 |
N_VIDEO_FRAMES = 5
|
| 15 |
|
| 16 |
IMAGE_TYPES = {"image/jpeg", "image/jpg", "image/png", "image/webp"}
|
|
|
|
| 9 |
from .model import load_detector, predict_image
|
| 10 |
from .video import sample_frames
|
| 11 |
|
| 12 |
+
MAX_IMAGE_SIZE_MB = 50
|
| 13 |
+
MAX_VIDEO_SIZE_MB = 300
|
| 14 |
N_VIDEO_FRAMES = 5
|
| 15 |
|
| 16 |
IMAGE_TYPES = {"image/jpeg", "image/jpg", "image/png", "image/webp"}
|
app/static/index.html
CHANGED
|
@@ -251,14 +251,14 @@
|
|
| 251 |
|
| 252 |
function getVerdict(aiScore, mediaType) {
|
| 253 |
const T = t();
|
| 254 |
-
if (aiScore >
|
| 255 |
return {
|
| 256 |
verdict: mediaType === "video" ? T.verdict_ai_video : T.verdict_ai_image,
|
| 257 |
advice: T.advice_ai,
|
| 258 |
tone: "ai",
|
| 259 |
};
|
| 260 |
}
|
| 261 |
-
if (aiScore >
|
| 262 |
return {
|
| 263 |
verdict: mediaType === "video" ? T.verdict_uncertain_video : T.verdict_uncertain_image,
|
| 264 |
advice: T.advice_uncertain,
|
|
@@ -324,7 +324,7 @@
|
|
| 324 |
showError(t().error_type);
|
| 325 |
return;
|
| 326 |
}
|
| 327 |
-
const maxMB = isVideo ?
|
| 328 |
if (file.size / (1024 * 1024) > maxMB) {
|
| 329 |
showError(t().error_size);
|
| 330 |
return;
|
|
|
|
| 251 |
|
| 252 |
function getVerdict(aiScore, mediaType) {
|
| 253 |
const T = t();
|
| 254 |
+
if (aiScore > 0.60) {
|
| 255 |
return {
|
| 256 |
verdict: mediaType === "video" ? T.verdict_ai_video : T.verdict_ai_image,
|
| 257 |
advice: T.advice_ai,
|
| 258 |
tone: "ai",
|
| 259 |
};
|
| 260 |
}
|
| 261 |
+
if (aiScore > 0.30) {
|
| 262 |
return {
|
| 263 |
verdict: mediaType === "video" ? T.verdict_uncertain_video : T.verdict_uncertain_image,
|
| 264 |
advice: T.advice_uncertain,
|
|
|
|
| 324 |
showError(t().error_type);
|
| 325 |
return;
|
| 326 |
}
|
| 327 |
+
const maxMB = isVideo ? 300 : 50;
|
| 328 |
if (file.size / (1024 * 1024) > maxMB) {
|
| 329 |
showError(t().error_size);
|
| 330 |
return;
|