Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -172,9 +172,7 @@ tag_model = ram(pretrained='preset/models/ram_swin_large_14m.pth',
|
|
172 |
tag_model.eval()
|
173 |
tag_model.to(device, dtype=weight_dtype)
|
174 |
|
175 |
-
def preprocess_image(
|
176 |
-
input_image: Image.Image
|
177 |
-
)
|
178 |
input_image = input_image.resize((256, 256), Image.Resampling.BILINEAR)
|
179 |
|
180 |
return input_image
|
|
|
172 |
tag_model.eval()
|
173 |
tag_model.to(device, dtype=weight_dtype)
|
174 |
|
175 |
+
def preprocess_image(input_image: Image.Image):
|
|
|
|
|
176 |
input_image = input_image.resize((256, 256), Image.Resampling.BILINEAR)
|
177 |
|
178 |
return input_image
|