ongkn commited on
Commit
c7a063d
·
unverified ·
1 Parent(s): 4534e51

fuckin' copilot tried rickrolling me

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -45,14 +45,11 @@ def classify_image(input):
45
  top_k=10
46
  )
47
  gradCamImage = gradCam.run_grad_cam_on_image(model=model,
48
- target_layer=targetLayerGradCam,
49
- classifier=model.classifier,
50
- img_pil=faceResized,
51
- img_tensor=tensorResized,
52
- reshape_transform=gradCam.reshape_transform_vit_huggingface,
53
- n_components=5,
54
- top_k=10
55
- )
56
  result = pipe(face)
57
  return result[0]["label"], result[0]["score"], face, dffImage, gradCamImage
58
 
 
45
  top_k=10
46
  )
47
  gradCamImage = gradCam.run_grad_cam_on_image(model=model,
48
+ target_layer=targetLayerGradCam,
49
+ targets_for_gradcam=targetsForGradCam,
50
+ input_tensor=tensorResized,
51
+ input_image=faceResized,
52
+ reshape_transform=gradCam.reshape_transform_vit_huggingface)
 
 
 
53
  result = pipe(face)
54
  return result[0]["label"], result[0]["score"], face, dffImage, gradCamImage
55