Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+from PIL import Image
 
 pipe = pipeline(
     model="alfiannajih/trash-classification",
@@ -7,6 +8,8 @@ pipe = pipeline(
 )
 
 def predict(image):
+    image = Image.fromarray(image)
+        
     pred = pipe(image)
 
     return pred