fix alpha
Browse files
JTP_PILOT/inference_gradio.py
CHANGED
|
@@ -137,7 +137,7 @@ sorted_tag_score = {}
|
|
| 137 |
|
| 138 |
def run_classifier(image, threshold):
|
| 139 |
global sorted_tag_score
|
| 140 |
-
img = image.convert('
|
| 141 |
tensor = transform(img).unsqueeze(0)
|
| 142 |
|
| 143 |
if torch.cuda.is_available():
|
|
|
|
| 137 |
|
| 138 |
def run_classifier(image, threshold):
|
| 139 |
global sorted_tag_score
|
| 140 |
+
img = image.convert('RGBA')
|
| 141 |
tensor = transform(img).unsqueeze(0)
|
| 142 |
|
| 143 |
if torch.cuda.is_available():
|