Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4e72d7a
1
Parent(s):
43d7e61
update
Browse files
app.py
CHANGED
|
@@ -44,6 +44,7 @@ def preprocess_image(image: Image.Image) -> Image.Image:
|
|
| 44 |
Returns:
|
| 45 |
Image.Image: The preprocessed image.
|
| 46 |
"""
|
|
|
|
| 47 |
processed_image = pipeline.preprocess_image(image)
|
| 48 |
return processed_image
|
| 49 |
|
|
|
|
| 44 |
Returns:
|
| 45 |
Image.Image: The preprocessed image.
|
| 46 |
"""
|
| 47 |
+
print(torch.cuda.get_device_name())
|
| 48 |
processed_image = pipeline.preprocess_image(image)
|
| 49 |
return processed_image
|
| 50 |
|