Spaces:

jimmyhnn
/

Garbage-Classifcation

Sleeping

Jimmy N commited on Oct 12, 2024

Commit

f7f64f8

1 Parent(s): 64b9b32

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,6 +6,8 @@ from PIL import Image
 pipe = pipeline("image-classification", model="yangy50/garbage-classification")
 def classify_image(image):
     results = pipe(image)
     return {result['label']: result['score'] for result in results}

 pipe = pipeline("image-classification", model="yangy50/garbage-classification")
 def classify_image(image):
+    # Ensure image is in RGB mode
+    image = image.convert("RGB")
     results = pipe(image)
     return {result['label']: result['score'] for result in results}