bwconrad commited on
Commit
fad6065
·
1 Parent(s): 31e0ac6

Update description

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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 Model",
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()