Spaces:
Build error
Build error
Update description
Browse files
app.py
CHANGED
@@ -62,10 +62,16 @@ labels = pd.read_csv("classid_classname.csv", names=["id", "name"])["name"].toli
|
|
62 |
labels = [l.replace("_", " ").title() for l in labels] # Remove _ and capitalize
|
63 |
|
64 |
# Run app
|
65 |
-
description = """
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
|
67 |
app = gr.Interface(
|
68 |
-
title="Classification
|
69 |
description=description,
|
70 |
fn=run,
|
71 |
inputs=[gr.Image(type="pil", tool="select"), gr.Checkbox(label="auto_crop")],
|
@@ -76,6 +82,5 @@ app = gr.Interface(
|
|
76 |
["futaba.jpg"],
|
77 |
["yotsuba.jpg"],
|
78 |
],
|
79 |
-
|
80 |
)
|
81 |
-
app.launch()
|
|
|
62 |
labels = [l.replace("_", " ").title() for l in labels] # Remove _ and capitalize
|
63 |
|
64 |
# Run app
|
65 |
+
description = """
|
66 |
+
A character classification model trained on the DAF:re dataset which consists of 3263 characters from anime, manga and video game series.
|
67 |
+
A list of all characters can be found [here](https://github.com/bwconrad/dafre/blob/main/app/classid_classname.csv).
|
68 |
+
|
69 |
+
The model is trained and performs best on head and shoulder portrait images.
|
70 |
+
Users can manually crop images through the UI or check the `auto_crop` box to let a face detection model do the cropping.
|
71 |
+
"""
|
72 |
|
73 |
app = gr.Interface(
|
74 |
+
title="Anime Character Classification",
|
75 |
description=description,
|
76 |
fn=run,
|
77 |
inputs=[gr.Image(type="pil", tool="select"), gr.Checkbox(label="auto_crop")],
|
|
|
82 |
["futaba.jpg"],
|
83 |
["yotsuba.jpg"],
|
84 |
],
|
|
|
85 |
)
|
86 |
+
app.launch()
|