ahh

app.py

@@ -24,7 +24,8 @@ class UPSC(nn.Module):
         return self.model(x)
 
 model = UPSC().to(device)
-model.load_state_dict(torch.load("upscaling.pth", weights_only=True,map_location=device))
+state = torch.load("upscaling.pth", weights_only=True,map_location=device)
+model.load_state_dict(state)
 model.eval()
 
 def task(img):