Medical_Image_Super_Resolution

Sleeping

suyoyog commited on Feb 28, 2024

Commit

152a480

verified ·

1 Parent(s): 9192bcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -399,7 +399,7 @@ def preprocess_input(input_image):
 def test_single_image( input_image):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     # generator_path = './pre_trained_model_064.pt'
-    generator_path = './MedSRGAN_gene_032.pt'

 def test_single_image( input_image):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     # generator_path = './pre_trained_model_064.pt'
+    generator_path = './MedSRGAN_gene_016.pt'