Sonic_Character_tagger

Runtime error

Metal079 commited on Jan 26, 2023

Commit

bb67038

1 Parent(s): aeabc6f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-pipe_aesthetic = pipeline("image-classification", "cafeai/cafe_aesthetic")
 def aesthetic(input_img):
     data = pipe_aesthetic(input_img, top_k=2)
     final = {}

 import gradio as gr
 from transformers import pipeline
+pipe_aesthetic = pipeline("image-classification", "./sonic")
 def aesthetic(input_img):
     data = pipe_aesthetic(input_img, top_k=2)
     final = {}