violence_api_issues_test

Sleeping

lelouch0110 commited on Nov 11, 2024

Commit

23b41f2

verified ·

1 Parent(s): 22c0b35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,6 +60,8 @@ def classify_image(inp):
               logging.warning("4")
               image = inp
     # Resize image to 224x224
     image = image.resize((224, 224), Image.Resampling.LANCZOS)

               logging.warning("4")
               image = inp
+    if image.mode != "RGB":
+        image = image.convert("RGB")
     # Resize image to 224x224
     image = image.resize((224, 224), Image.Resampling.LANCZOS)