Spaces:
Running
on
Zero
Running
on
Zero
fix alpha
Browse files
app.py
CHANGED
|
@@ -152,7 +152,7 @@ sorted_tag_score = {}
|
|
| 152 |
@spaces.GPU(duration=5)
|
| 153 |
def run_classifier(image, threshold):
|
| 154 |
global sorted_tag_score
|
| 155 |
-
img = image.convert('
|
| 156 |
tensor = transform(img).unsqueeze(0)
|
| 157 |
|
| 158 |
with torch.no_grad():
|
|
|
|
| 152 |
@spaces.GPU(duration=5)
|
| 153 |
def run_classifier(image, threshold):
|
| 154 |
global sorted_tag_score
|
| 155 |
+
img = image.convert('RGBA')
|
| 156 |
tensor = transform(img).unsqueeze(0)
|
| 157 |
|
| 158 |
with torch.no_grad():
|