Spaces:
Build error
Build error
Commit
·
06624f9
0
Parent(s):
Duplicate from Dinoking/Garbage-Classifier-V6
Browse filesCo-authored-by: Aditya <[email protected]>
- .python-version +1 -0
- .gitattributes +31 -0
- README.md +14 -0
- app.py +28 -0
- battery.jpg +0 -0
- cardboard.jpeg +0 -0
- clothes.jpeg +0 -0
- glass.jpg +0 -0
- metal.jpg +0 -0
- model804.h5 +3 -0
- plastic.jpg +0 -0
- requirements.txt +3 -0
- shoes.jpg +0 -0
.python-version
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
3.7.6
|
.gitattributes
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
23 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
26 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.zstandard filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
title: Garbage Classifier V6
|
3 |
+
emoji: 😻
|
4 |
+
colorFrom: gray
|
5 |
+
colorTo: gray
|
6 |
+
sdk: gradio
|
7 |
+
sdk_version: 3.1.4
|
8 |
+
app_file: app.py
|
9 |
+
pinned: false
|
10 |
+
python_version: 3.7.6
|
11 |
+
duplicated_from: Dinoking/Garbage-Classifier-V6
|
12 |
+
---
|
13 |
+
|
14 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import tensorflow as tf
|
3 |
+
import numpy as np
|
4 |
+
from PIL import Image
|
5 |
+
import tensorflow.keras as keras
|
6 |
+
import keras.applications.xception as xception
|
7 |
+
from tensorflow.keras.models import load_model
|
8 |
+
|
9 |
+
# load model
|
10 |
+
model = load_model('model804.h5')
|
11 |
+
|
12 |
+
classnames = ['battery','cardboard','clothes','food','glass','medical','metal','paper','plastic','shoes']
|
13 |
+
|
14 |
+
|
15 |
+
|
16 |
+
def predict_image(img):
|
17 |
+
img_4d=img.reshape(-1,320, 320,3)
|
18 |
+
prediction=model.predict(img_4d)[0]
|
19 |
+
return {classnames[i]: float(prediction[i]) for i in range(10)}
|
20 |
+
|
21 |
+
image = gr.inputs.Image(shape=(320, 320))
|
22 |
+
label = gr.outputs.Label(num_top_classes=3)
|
23 |
+
enable_queue=True
|
24 |
+
examples = ['battery.jpg','cardboard.jpeg','clothes.jpeg','glass.jpg','metal.jpg','plastic.jpg','shoes.jpg']
|
25 |
+
article="<p style='text-align: center'>Made by Aditya Narendra with 🖤</p>"
|
26 |
+
|
27 |
+
gr.Interface(fn=predict_image, inputs=image, title="Garbage Classifier",
|
28 |
+
description="This is a Garbage Classification Model Trained using Xception Net on DS11 Mod(Seg10 V4).Deployed to Hugging Faces using Gradio.",outputs=label,article=article,enable_queue=enable_queue,examples=examples,interpretation='default').launch(share="True")
|
battery.jpg
ADDED
![]() |
cardboard.jpeg
ADDED
![]() |
clothes.jpeg
ADDED
![]() |
glass.jpg
ADDED
![]() |
metal.jpg
ADDED
![]() |
model804.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:adba5e3a50ae004a826f1ff3df5c5841b7ec59e781cff0fe7a9ee158d8724fb2
|
3 |
+
size 83963016
|
plastic.jpg
ADDED
![]() |
requirements.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
numpy
|
2 |
+
Pillow
|
3 |
+
tensorflow
|
shoes.jpg
ADDED
![]() |