Hugging Face's logo

Spaces:
zihaoz96
/
shark-classifier
Runtime error

App Files Community
zihaoz96 commited on
Commit
c7b6585
•
1 Parent(s): 5ecfb97

update

Browse files
Files changed (7) hide show
  1. app.py +2 -1
  2. models/DenseNet/best_model.pth +3 -0
  3. models/DenseNet/config.json +39 -0
  4. models/DenseNet/model.h5 +3 -0
  5. models/VGG16/best_model.pth +3 -0
  6. models/VGG16/config.json +39 -0
  7. models/{modelnet/best_model.h5 → mobilenet_v2/best_model.pth} +2 -2
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import gradio as gr
2
  import numpy as np
3
- from tensorflow.keras import models
4
 
5
  import tensorflow as tf
 
6
 
7
  models_name = [
8
  "VGG16",
 
1
  import gradio as gr
2
  import numpy as np
3
+ from PIL import Image
4
 
5
  import tensorflow as tf
6
+ from hugsvision.inference.TorchVisionClassifierInference import TorchVisionClassifierInference
7
 
8
  models_name = [
9
  "VGG16",
models/DenseNet/best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:392d9b51f1d2493f67a9ac0f78723593800969f662a6575a16b7ca66f5b57ce8
3
+ size 28498899
models/DenseNet/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "num_classes": 14,
3
+ "hidden_size": 1024,
4
+ "id2label": {
5
+ "0": "basking",
6
+ "1": "blacktip",
7
+ "2": "blue",
8
+ "3": "bull",
9
+ "4": "hammerhead",
10
+ "5": "lemon",
11
+ "6": "mako",
12
+ "7": "nurse",
13
+ "8": "sand_tiger",
14
+ "9": "thresher",
15
+ "10": "tiger",
16
+ "11": "whale",
17
+ "12": "white",
18
+ "13": "whitetip"
19
+ },
20
+ "label2id": {
21
+ "basking": "0",
22
+ "blacktip": "1",
23
+ "blue": "2",
24
+ "bull": "3",
25
+ "hammerhead": "4",
26
+ "lemon": "5",
27
+ "mako": "6",
28
+ "nurse": "7",
29
+ "sand_tiger": "8",
30
+ "thresher": "9",
31
+ "tiger": "10",
32
+ "whale": "11",
33
+ "white": "12",
34
+ "whitetip": "13"
35
+ },
36
+ "architectures": [
37
+ "densenet121"
38
+ ]
39
+ }
models/DenseNet/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e4649a9bcb5b2d582da51237755cea1950f7abe8c9d59d41eb15a8ad0f4e1a9
3
+ size 86078232
models/VGG16/best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35d59f3cc484f1490d24b7947e3e82730f1cac6f59da407be789fff44f3afba0
3
+ size 537289349
models/VGG16/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "num_classes": 14,
3
+ "hidden_size": 4096,
4
+ "id2label": {
5
+ "0": "basking",
6
+ "1": "blacktip",
7
+ "2": "blue",
8
+ "3": "bull",
9
+ "4": "hammerhead",
10
+ "5": "lemon",
11
+ "6": "mako",
12
+ "7": "nurse",
13
+ "8": "sand_tiger",
14
+ "9": "thresher",
15
+ "10": "tiger",
16
+ "11": "whale",
17
+ "12": "white",
18
+ "13": "whitetip"
19
+ },
20
+ "label2id": {
21
+ "basking": "0",
22
+ "blacktip": "1",
23
+ "blue": "2",
24
+ "bull": "3",
25
+ "hammerhead": "4",
26
+ "lemon": "5",
27
+ "mako": "6",
28
+ "nurse": "7",
29
+ "sand_tiger": "8",
30
+ "thresher": "9",
31
+ "tiger": "10",
32
+ "whale": "11",
33
+ "white": "12",
34
+ "whitetip": "13"
35
+ },
36
+ "architectures": [
37
+ "vgg16"
38
+ ]
39
+ }
models/{modelnet/best_model.h5 → mobilenet_v2/best_model.pth} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e7c1ec8dba0efbf87bd07ad89555987bd7be807efbc77d42cab2d0c75a45d6a0
3
- size 9556672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:650c8c31d438caf26c768908e204a0fd8ee48af3f22d157e6eb88237dc103828
3
+ size 9235021