ibrim commited on
Commit
764fcd0
·
verified ·
1 Parent(s): e4057e4

Update gradio_utils.py

Browse files
Files changed (1) hide show
  1. gradio_utils.py +1 -1
gradio_utils.py CHANGED
@@ -88,7 +88,7 @@ class LitCIFAR(LightningModule):
88
  modelfin = LitCIFAR()
89
  lit_cifar_instance = LitCIFAR()
90
  # Load the state dictionary from the checkpoint file
91
- modelfin.load_state_dict(torch.load("ibrim/TSAIGradcam/model.ckpt"))
92
  # Set the model to evaluation mode
93
  modelfin.eval()
94
  # 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
 
88
  modelfin = LitCIFAR()
89
  lit_cifar_instance = LitCIFAR()
90
  # Load the state dictionary from the checkpoint file
91
+ modelfin.load_state_dict(torch.load("TSAIGradcam/model.ckpt"))
92
  # Set the model to evaluation mode
93
  modelfin.eval()
94
  # 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