fix: path

app.py

@@ -20,7 +20,7 @@ def process_image(input_img,score,model_path,gamma=1.0,alpha_s=1.0,alpha_i=1.0):
     if model_path is None:
         return input_img,"Please choose a model weights."
     torch.set_grad_enabled(False)
-    eval_net.from_pretrained(model_path)
+    eval_net.from_pretrained("HVI-CIDNet-"+model_path)
     # eval_net.load_state_dict(torch.load(os.path.join(directory,model_path), map_location=lambda storage, loc: storage))
     eval_net.eval()
     
@@ -49,16 +49,16 @@ def process_image(input_img,score,model_path,gamma=1.0,alpha_s=1.0,alpha_i=1.0):
 
 directory = "weights"
 pth_files = [
-    'HVI-CIDNet-Generalization',
-    'HVI-CIDNet-Sony-Total-Dark',
-    'HVI-CIDNet-LOL-Blur',
-    'HVI-CIDNet-SICE',
-    'HVI-CIDNet-LOLv2-real-bestSSIM',
-    'HVI-CIDNet-LOLv2-real-bestPSNR',
-    'HVI-CIDNet-LOLv2-syn-wperc',
-    'HVI-CIDNet-LOLv2-syn-woperc',
-    'HVI-CIDNet-LOLv1-wperc',
-    'HVI-CIDNet-LOLv1-woperc'
+    'Generalization',
+    'Sony-Total-Dark',
+    'LOL-Blur',
+    'SICE',
+    'LOLv2-real-bestSSIM',
+    'LOLv2-real-bestPSNR',
+    'LOLv2-syn-wperc',
+    'LOLv2-syn-woperc',
+    'LOLv1-wperc',
+    'LOLv1-woperc'
 ]
 
 
@@ -67,7 +67,7 @@ interface = gr.Interface(
     inputs=[
         gr.Image(label="Low-light Image", type="pil"),
         gr.Radio(choices=['Yes','No'],label="Image Score",info="Calculate NIQE, default is \"No\"."),
-        gr.Radio(choices=pth_files2,label="Model Weights",info="Choose your model. The best models are \"SICE.pth\" and \"generalization.pth\"."),
+        gr.Radio(choices=pth_files,label="Model Weights",info="Choose your model. The best models are \"SICE.pth\" and \"generalization.pth\"."),
         gr.Slider(0.1,5,label="gamma curve",step=0.01,value=1.0, info="Lower is lighter, best range is [0.5,2.5]."),
         gr.Slider(0,2,label="Alpha-s",step=0.01,value=1.0, info="Higher is more saturated."),
         gr.Slider(0.1,2,label="Alpha-i",step=0.01,value=1.0, info="Higher is lighter.")