manmeet3591 commited on
Commit
60da226
·
verified ·
1 Parent(s): 4b697f3

Update example.py

Browse files
Files changed (1) hide show
  1. example.py +1 -1
example.py CHANGED
@@ -1,6 +1,6 @@
1
  # example.py
2
  import torch
3
- from model import MeteoGAN # Import your model class
4
 
5
  # Load the pre-trained model
6
  model = MeteoGAN(in_channels=3, out_channels=3, upscale=4)
 
1
  # example.py
2
  import torch
3
+ from model import SRResNet # Import your model class
4
 
5
  # Load the pre-trained model
6
  model = MeteoGAN(in_channels=3, out_channels=3, upscale=4)