Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
+from PIL import Image
 from transformers import pipeline
 
 pipeline = pipeline(
-    "image-classification", model="fauzanardh/anime-time-beit_classifier"
+    "image-classification",
+    model="fauzanardh/anime-time-beit_classifier",
 )
 
 
-def classify_image(input_img):
+def classify_image(input_img: Image.Image) -> dict:
     data = pipeline(input_img, top_k=4)
     out = {}
     for d in data:
@@ -20,4 +22,5 @@ demo = gr.Interface(
     outputs=gr.Label(label="Prediction"),
 )
 
+demo.queue()
 demo.launch()