waifu demo

app.py

@@ -19,4 +19,13 @@ def style(input_img):
     return final
 demo_style = gr.Interface(fn=style, inputs=gr.Image(type="pil"), outputs=gr.Label(label="style"))
 
-gr.Parallel(demo_aesthetic, demo_style).launch()
+pipe_waifu = pipeline("image-classification", "cafeai/cafe_waifu")
+def waifu(input_img):
+    data = pipe_waifu(input_img, top_k=5)
+    final = {}
+    for d in data:
+        final[d["label"]] = d["score"]
+    return final
+demo_waifu = gr.Interface(fn=waifu, inputs=gr.Image(type="pil"), outputs=gr.Label(label="waifu"))
+
+gr.Parallel(demo_aesthetic, demo_style, demo_waifu).launch()