Spaces:
Sleeping
Sleeping
Update gradio_utils.py
Browse files- 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("
|
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
|