Spaces:

xiaosu-zhu
/

McQuic

Runtime error

xiaosu-zhu commited on Mar 30, 2022

Commit

5197152

1 Parent(s): ba2bd35

Auto deploy

Files changed (1) hide show

stCompressService.py CHANGED Viewed

@@ -47,8 +47,7 @@ def compressImage(image: torch.Tensor, model: BaseCompressor, crop: bool) -> Fil
     image = (image - 0.5) * 2
     with model.readyForCoding() as cdfs:
-        codes, size = model.encode(images)
-        binaries, headers = model.compress(self._encoder, codes, size, cdfs)
     return File(headers[0], binaries[0])

     image = (image - 0.5) * 2
     with model.readyForCoding() as cdfs:
+        codes, binaries, headers = model.compress(image[None, ...], cdfs)
     return File(headers[0], binaries[0])