English
Inference Endpoints
garg-aayush commited on
Commit
68853b7
1 Parent(s): e0d40ee

weight path fix

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -25,8 +25,8 @@ class EndpointHandler:
25
  self.model = RealESRGANer(
26
  scale=4, # Scale factor for the model
27
  # Path to the pre-trained model weights
28
- # model_path=f"/repository/weights/Real-ESRGAN-x4plus.pth",
29
- model_path=f"/workspace/real-esrgan/weights/Real-ESRGAN-x4plus.pth",
30
  # Initialize the RRDBNet model architecture with specified parameters
31
  model= RRDBNet(num_in_ch=3,
32
  num_out_ch=3,
 
25
  self.model = RealESRGANer(
26
  scale=4, # Scale factor for the model
27
  # Path to the pre-trained model weights
28
+ model_path=f"/repository/weights/Real-ESRGAN-x4plus.pth",
29
+ # model_path=f"/workspace/real-esrgan/weights/Real-ESRGAN-x4plus.pth",
30
  # Initialize the RRDBNet model architecture with specified parameters
31
  model= RRDBNet(num_in_ch=3,
32
  num_out_ch=3,