vicliv commited on
Commit
419e73d
·
1 Parent(s): d2354a4

added softmax temperature for better uncertainty calibration

Browse files
Files changed (2) hide show
  1. app/main.py +2 -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 = 20
13
- MAX_VIDEO_SIZE_MB = 100
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 >= 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,7 +324,7 @@
324
  showError(t().error_type);
325
  return;
326
  }
327
- const maxMB = isVideo ? 100 : 20;
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;