Spaces:

Cidaut
/

DarkIR

Running

danifei commited on Dec 18, 2024

Commit

f603c60

verified ·

1 Parent(s): 5c52748

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ tensor_to_pil = transforms.ToPILImage()
 network = 'DarkIR'
-PATH_MODEL = './darkir_v2real+lsrw.pt'
 model = DarkIR(img_channel=3,
                     width=32,

 network = 'DarkIR'
+PATH_MODEL = './DarkIR_384.pt'
 model = DarkIR(img_channel=3,
                     width=32,