JointTaggerProject-Inference-Beta

Running

drhead commited on Jul 7, 2024

Commit

e6d942c

1 Parent(s): 4d75269

fix alpha

Files changed (1) hide show

app.py CHANGED Viewed

@@ -152,7 +152,7 @@ sorted_tag_score = {}
 @spaces.GPU(duration=5)
 def run_classifier(image, threshold):
     global sorted_tag_score
-    img = image.convert('RGB')
     tensor = transform(img).unsqueeze(0)
     with torch.no_grad():

 @spaces.GPU(duration=5)
 def run_classifier(image, threshold):
     global sorted_tag_score
+    img = image.convert('RGBA')
     tensor = transform(img).unsqueeze(0)
     with torch.no_grad():