Serkan Ozturk
commited on
Commit
·
65c97a3
1
Parent(s):
c07b674
gsfglşsdkif1
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -42,7 +42,7 @@ class EndpointHandler():
|
|
| 42 |
upscaled = self.model(img)
|
| 43 |
upscaled = upscaled.squeeze(0).permute(1,2,0)
|
| 44 |
upscaled = upscaled.numpy()
|
| 45 |
-
upscaled = np.clip(upscaled
|
| 46 |
|
| 47 |
|
| 48 |
pil = Image.fromarray(upscaled)
|
|
|
|
| 42 |
upscaled = self.model(img)
|
| 43 |
upscaled = upscaled.squeeze(0).permute(1,2,0)
|
| 44 |
upscaled = upscaled.numpy()
|
| 45 |
+
upscaled = np.clip(upscaled, 0, 255).astype(np.uint8)
|
| 46 |
|
| 47 |
|
| 48 |
pil = Image.fromarray(upscaled)
|