anky2002 commited on
Commit
4529fa9
·
verified ·
1 Parent(s): 57eeb27

Upload agents/model_agent.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. agents/model_agent.py +2 -2
agents/model_agent.py CHANGED
@@ -238,8 +238,8 @@ ALL_TESTS=[m01_fft_grid_8x8,m02_fft_grid_16x16,m03_spectral_slope,m04_diffusion_
238
  m09_upsampling_stride,m10_patch_diversity,m11_color_consistency,m12_spectral_rolloff_shape,
239
  m13_texture_repetition,m14_highfreq_noise_structure,m15_phase_coherence]
240
 
241
- def run_model_agent(img):
242
  from agents.utils import run_agent_tests
243
  from agents.optical_agent import AgentEvidence
244
- findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Generative Model Agent")
245
  return AgentEvidence("Generative Model Agent",np.clip(avg,-1,1),conf,fail,rat,findings)
 
238
  m09_upsampling_stride,m10_patch_diversity,m11_color_consistency,m12_spectral_rolloff_shape,
239
  m13_texture_repetition,m14_highfreq_noise_structure,m15_phase_coherence]
240
 
241
+ def run_model_agent(img, modality_adjustments=None):
242
  from agents.utils import run_agent_tests
243
  from agents.optical_agent import AgentEvidence
244
+ findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Generative Model Agent", modality_adjustments)
245
  return AgentEvidence("Generative Model Agent",np.clip(avg,-1,1),conf,fail,rat,findings)