CrypticMonkey3 commited on
Commit
ba73931
Β·
verified Β·
1 Parent(s): a8ba371

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -12
app.py CHANGED
@@ -71,19 +71,23 @@ def analyze_video(video):
71
 
72
  ai_scores = probs[:, ai_class_index].tolist()
73
 
74
- # Verdict logic
75
- ai_conf_threshold = 0.7
76
- min_ai_frames = 3
77
  avg_score = sum(ai_scores) / len(ai_scores)
78
- confident_ai_frames = sum(1 for score in ai_scores if score >= ai_conf_threshold)
79
-
80
- if confident_ai_frames >= min_ai_frames:
81
- verdict = "Video is likely AI-generated."
82
- elif avg_score >= 0.5:
83
- verdict = "Uncertain β€” some signs of AI generation but not enough evidence."
84
- else:
85
- verdict = "Video is likely REAL."
86
-
 
 
 
 
 
 
 
87
  # Output results
88
  results = []
89
  for frame_idx, pred, ai_score in zip(frame_numbers, predictions, ai_scores):
 
71
 
72
  ai_scores = probs[:, ai_class_index].tolist()
73
 
 
 
 
74
  avg_score = sum(ai_scores) / len(ai_scores)
75
+ total_frames = len(ai_scores)
76
+ ai_frame_count = sum(1 for score in ai_scores if score >= 0.5) # Soft threshold for "suspicious" frame
77
+ ai_frame_ratio = ai_frame_count / total_frames
78
+
79
+ # Refined verdict based on AI frame ratio and confidence
80
+ if ai_frame_ratio < 0.25:
81
+ verdict = "βœ… Very likely real β€” no strong signs of AI generation."
82
+ elif 0.25 <= ai_frame_ratio < 0.5:
83
+ verdict = "⚠️ Mostly real β€” a few signs could raise suspicion."
84
+ elif ai_frame_ratio >= 0.5:
85
+ if avg_score >= 0.85:
86
+ verdict = "🚨 Definitely AI-generated β€” high confidence across many frames."
87
+ elif avg_score >= 0.75:
88
+ verdict = "⚠️ Likely AI-generated β€” multiple signs suggest manipulation."
89
+ else:
90
+ verdict = "πŸ€” Unclear β€” the video shows mixed signals."
91
  # Output results
92
  results = []
93
  for frame_idx, pred, ai_score in zip(frame_numbers, predictions, ai_scores):