anime-ai-detect / app.py
xiaoguolizi's picture
Duplicate from saltacc/anime-ai-detect
440ce25
raw
history blame contribute delete
420 Bytes
import gradio as gr
from transformers import pipeline
detection_pipeline = pipeline("image-classification", "saltacc/anime-ai-detect")
def detect(img):
print(img)
output = detection_pipeline(img, top_k=2)
final = {}
for d in output:
final[d["label"]] = d["score"]
return final
iface = gr.Interface(fn=detect, inputs=gr.Image(type="pil"), outputs=gr.Label(label="result"))
iface.launch()