Update example.py
Browse files- example.py +1 -1
example.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# example.py
|
| 2 |
import torch
|
| 3 |
-
from model import
|
| 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)
|