gr.selectdata can't be pickled fix

app.py

@@ -239,15 +239,15 @@ def clear_image():
     return "", {}, None, {}, None
 
 @spaces.GPU(duration=5)
-def cam_inference(img, threshold, alpha, evt: gr.SelectData):
-    target_tag_index = tags[evt.value]
+def cam_inference(img, threshold, alpha, selected_tag: str):
+    target_tag_index = tags[selected_tag] 
     tensor = transform(img).unsqueeze(0)
 
+
     if torch.cuda.is_available():
         tensor = tensor.to(device, dtype=torch.float16)
     else:
         tensor = tensor.to(device)
-    
     tensor.requires_grad_()
 
     gradients = {}
@@ -263,7 +263,7 @@ def cam_inference(img, threshold, alpha, evt: gr.SelectData):
     handle_backward = model.norm.register_full_backward_hook(hook_backward)
 
     probits = model(tensor)[0]
- 
+
     model.zero_grad()
     probits[target_tag_index].backward(retain_graph=True)
 
@@ -470,7 +470,7 @@ with gr.Blocks(css=custom_css) as demo:
 
             label_box.select(
                 fn=cam_inference,
-                inputs=[original_image_state, cam_slider, alpha_slider],
+                inputs=[original_image_state, cam_slider, alpha_slider, label_box],
                 outputs=[image, cam_state],
                 show_progress='minimal'
             )