Zilong-Zhang003 commited on
Commit
cbe80b8
·
1 Parent(s): 528fad1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from restormerRFR_arch import RestormerRFR
17
  from dino_feature_extractor import DinoFeatureModule
18
 
19
  WEIGHT_REPO_ID = "233zzl/RAM_plus_plus"
20
- WEIGHT_FILENAME = "7task/RestormerRFR.pth"
21
  MODEL_NAME = "RestormerRFR"
22
 
23
  def get_device():
 
17
  from dino_feature_extractor import DinoFeatureModule
18
 
19
  WEIGHT_REPO_ID = "233zzl/RAM_plus_plus"
20
+ WEIGHT_FILENAME = "ram_plus/7task/RestormerRFR.pth"
21
  MODEL_NAME = "RestormerRFR"
22
 
23
  def get_device():