irebmann commited on
Commit
555e888
·
verified ·
1 Parent(s): f2b467d

changed code to localfiles=true to fix error

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,15 +8,15 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
9
  # ---- Model IDs / paths ----
10
  PRETRAINED_ID = "caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr" # crystal clear
11
- FINETUNED_ID = "swin2sr_div2k_finetuned_1000steps" # smooth (local folder in repo)
12
 
13
  # ---- Load processors ----
14
  processor_pre = AutoImageProcessor.from_pretrained(PRETRAINED_ID)
15
- processor_ft = AutoImageProcessor.from_pretrained(FINETUNED_ID)
16
 
17
  # ---- Load models ----
18
  model_pre = Swin2SRForImageSuperResolution.from_pretrained(PRETRAINED_ID).to(device)
19
- model_ft = Swin2SRForImageSuperResolution.from_pretrained(FINETUNED_ID).to(device)
20
 
21
  model_pre.eval()
22
  model_ft.eval()
 
8
 
9
  # ---- Model IDs / paths ----
10
  PRETRAINED_ID = "caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr" # crystal clear
11
+ FINETUNED_ID = "./swin2sr_div2k_finetuned_1000steps" # smooth (local folder in repo)
12
 
13
  # ---- Load processors ----
14
  processor_pre = AutoImageProcessor.from_pretrained(PRETRAINED_ID)
15
+ processor_ft = AutoImageProcessor.from_pretrained(FINETUNED_ID, local_files_only=True)
16
 
17
  # ---- Load models ----
18
  model_pre = Swin2SRForImageSuperResolution.from_pretrained(PRETRAINED_ID).to(device)
19
+ model_ft = Swin2SRForImageSuperResolution.from_pretrained(FINETUNED_ID, local_files_only=True).to(device)
20
 
21
  model_pre.eval()
22
  model_ft.eval()