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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ 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)
 
8
 
9
  # ---- Model IDs / paths ----
10
  PRETRAINED_ID = "caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr" # crystal clear
11
+ FINETUNED_ID = "swin2sr_div2k_finetuned_x4_1000steps" # smooth (local folder in repo)
12
 
13
  # ---- Load processors ----
14
  processor_pre = AutoImageProcessor.from_pretrained(PRETRAINED_ID)