ibrim commited on
Commit
88738f4
·
verified ·
1 Parent(s): 497d8ca

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("submission_gradio/models/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("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