garg-aayush
commited on
Commit
•
68853b7
1
Parent(s):
e0d40ee
weight path fix
Browse files- 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 |
-
|
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,
|