Spaces:
Runtime error
Runtime error
set lineart detect_resolution to 384 (#3)
Browse files- set lineart detect_resolution to 384 (b5441a4871561c3b7d955db24bbd39ef049f6541)
Co-authored-by: Suraj Patil <[email protected]>
model.py
CHANGED
|
@@ -117,7 +117,7 @@ class LineartPreprocessor(Preprocessor):
|
|
| 117 |
return self.model.to(device)
|
| 118 |
|
| 119 |
def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
|
| 120 |
-
return self.model(image, detect_resolution=
|
| 121 |
|
| 122 |
|
| 123 |
class MidasPreprocessor(Preprocessor):
|
|
|
|
| 117 |
return self.model.to(device)
|
| 118 |
|
| 119 |
def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
|
| 120 |
+
return self.model(image, detect_resolution=384, image_resolution=1024)
|
| 121 |
|
| 122 |
|
| 123 |
class MidasPreprocessor(Preprocessor):
|