Update gradio_utils.py

gradio_utils.py

@@ -88,7 +88,7 @@ class LitCIFAR(LightningModule):
 modelfin = LitCIFAR()
 lit_cifar_instance = LitCIFAR()
 # Load the state dictionary from the checkpoint file
-modelfin.load_state_dict(torch.load("TSAIGradcam/model.ckpt"))
+modelfin.load_state_dict(torch.load("model.ckpt"))
 # Set the model to evaluation mode
 modelfin.eval()
 # If you need to use the model on a GPU, move the model to GPU after loading the state dict and setting it to eval mode