JointTaggerProject-Inference

Sleeping

drhead commited on Jul 7, 2024

Commit

8409352

verified ·

1 Parent(s): 08e34e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,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():