anky2002 commited on
Commit
9dd9d31
·
verified ·
1 Parent(s): 4529fa9

Upload agents/statistical_agent.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. agents/statistical_agent.py +2 -2
agents/statistical_agent.py CHANGED
@@ -304,8 +304,8 @@ ALL_TESTS=[t01_dct_kurtosis,t02_benford,t03_gradient_sparsity,t04_local_kurtosis
304
  t15_saturation_distribution,t16_luminance_gradient_ratio,t17_pixel_uniqueness,t18_global_entropy,
305
  t19_power_law_fit,t20_contrast_distribution,t21_joint_histogram,t22_run_length]
306
 
307
- def run_statistical_agent(img):
308
  from agents.utils import run_agent_tests
309
  from agents.optical_agent import AgentEvidence
310
- findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Statistical Priors Agent")
311
  return AgentEvidence("Statistical Priors Agent", np.clip(avg,-1,1), conf, fail, rat, findings)
 
304
  t15_saturation_distribution,t16_luminance_gradient_ratio,t17_pixel_uniqueness,t18_global_entropy,
305
  t19_power_law_fit,t20_contrast_distribution,t21_joint_histogram,t22_run_length]
306
 
307
+ def run_statistical_agent(img, modality_adjustments=None):
308
  from agents.utils import run_agent_tests
309
  from agents.optical_agent import AgentEvidence
310
+ findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Statistical Priors Agent", modality_adjustments)
311
  return AgentEvidence("Statistical Priors Agent", np.clip(avg,-1,1), conf, fail, rat, findings)