Update JTP_PILOT2/inference_gradio.py
Browse files
JTP_PILOT2/inference_gradio.py
CHANGED
|
@@ -156,7 +156,7 @@ sorted_tag_score = {}
|
|
| 156 |
|
| 157 |
def run_classifier(image, threshold):
|
| 158 |
global sorted_tag_score
|
| 159 |
-
img = image.convert('
|
| 160 |
tensor = transform(img).unsqueeze(0)
|
| 161 |
|
| 162 |
if torch.cuda.is_available():
|
|
|
|
| 156 |
|
| 157 |
def run_classifier(image, threshold):
|
| 158 |
global sorted_tag_score
|
| 159 |
+
img = image.convert('RGBA')
|
| 160 |
tensor = transform(img).unsqueeze(0)
|
| 161 |
|
| 162 |
if torch.cuda.is_available():
|