Spaces:

xPushpeshx
/

flare_removal_2.0

Sleeping

Pushpesh commited on Aug 11, 2025

Commit

92bae4b

1 Parent(s): afc145a

Fixed utils bug

Files changed (1) hide show

app/utils.py CHANGED Viewed

@@ -3,7 +3,7 @@ from PIL import Image
 import numpy as np
 device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model=torch.load('model/Unet_segformer.pth',map_location=device)
 def evaluate(model,image):
     model.eval()

 import numpy as np
 device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model=torch.load('model/model_epoch_49.pth',map_location=device)
 def evaluate(model,image):
     model.eval()