Light-amplification

Running on Zero

Fediory commited on 10 days ago

Commit

7e7be12

1 Parent(s): a7404ed

fix: path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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("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()

     if model_path is None:
         return input_img,"Please choose a model weights."
     torch.set_grad_enabled(False)
+    eval_net.from_pretrained("Fediory/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()