RedRocket
/

JointTaggerProject

Image Classification

Model card Files Files and versions Community

drhead commited on Jul 7

Commit

ac391c7

•

1 Parent(s): 8cba915

fix alpha

Files changed (1) hide show

JTP_PILOT/inference_gradio.py +1 -1

JTP_PILOT/inference_gradio.py CHANGED Viewed

@@ -137,7 +137,7 @@ sorted_tag_score = {}
 def run_classifier(image, threshold):
     global sorted_tag_score
-    img = image.convert('RGB')
     tensor = transform(img).unsqueeze(0)
     if torch.cuda.is_available():

 def run_classifier(image, threshold):
     global sorted_tag_score
+    img = image.convert('RGBA')
     tensor = transform(img).unsqueeze(0)
     if torch.cuda.is_available():