narugo commited on
Commit
3255ddb
1 Parent(s): 28f7419

dev(narugo): use better models

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ import gradio as gr
5
  from base import Classification
6
 
7
  apps = [
8
- Classification('Classification', 'deepghs/anime_classification', 'mobilenetv3_sce_dist'),
9
  Classification('Monochrome', 'deepghs/monochrome_detect', 'mobilenetv3_large_100_dist'),
10
- Classification('Completeness', 'deepghs/anime_completeness', 'caformer_s36_v0'),
11
  Classification('AI Check', 'deepghs/anime_ai_check', 'mobilenetv3_sce_dist'),
12
  Classification('AI Corrupt', 'deepghs/ai_image_corrupted', 'caformer_s36_v0_focal'),
13
  Classification('Rating', 'deepghs/anime_rating', 'mobilenetv3_sce_dist'),
@@ -15,7 +15,7 @@ apps = [
15
  Classification('Portrait Type', 'deepghs/anime_portrait', 'mobilenetv3_v0_dist'),
16
  Classification('Age of Style', 'deepghs/anime_style_ages', 'mobilenetv3_v0_dist'),
17
  Classification('Bangumi Portrait', 'deepghs/bangumi_char_type', 'mobilenetv3_v0_dist'),
18
- Classification('Is That Anime?', 'deepghs/anime_real_cls', 'mobilenetv3_v0_dist'),
19
  Classification('Teen', 'deepghs/anime_teen', 'caformer_s36_v0'),
20
  Classification('Character Skin', 'deepghs/anime_ch_skin_color', 'caformer_s36'),
21
  Classification('Character Hair Color', 'deepghs/anime_ch_hair_color', 'caformer_s36_v0'),
 
5
  from base import Classification
6
 
7
  apps = [
8
+ Classification('Classification', 'deepghs/anime_classification', 'mobilenetv3_v1.3_dist'),
9
  Classification('Monochrome', 'deepghs/monochrome_detect', 'mobilenetv3_large_100_dist'),
10
+ Classification('Completeness', 'deepghs/anime_completeness', 'mobilenetv3_v2.2_dist'),
11
  Classification('AI Check', 'deepghs/anime_ai_check', 'mobilenetv3_sce_dist'),
12
  Classification('AI Corrupt', 'deepghs/ai_image_corrupted', 'caformer_s36_v0_focal'),
13
  Classification('Rating', 'deepghs/anime_rating', 'mobilenetv3_sce_dist'),
 
15
  Classification('Portrait Type', 'deepghs/anime_portrait', 'mobilenetv3_v0_dist'),
16
  Classification('Age of Style', 'deepghs/anime_style_ages', 'mobilenetv3_v0_dist'),
17
  Classification('Bangumi Portrait', 'deepghs/bangumi_char_type', 'mobilenetv3_v0_dist'),
18
+ Classification('Is That Anime?', 'deepghs/anime_real_cls', 'mobilenetv3_v1.2_dist'),
19
  Classification('Teen', 'deepghs/anime_teen', 'caformer_s36_v0'),
20
  Classification('Character Skin', 'deepghs/anime_ch_skin_color', 'caformer_s36'),
21
  Classification('Character Hair Color', 'deepghs/anime_ch_hair_color', 'caformer_s36_v0'),