Taidat commited on
Commit
7a72b2f
·
verified ·
1 Parent(s): 418f916

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -29
app.py DELETED
@@ -1,29 +0,0 @@
1
- import gradio as gr
2
- import tensorflow as tf
3
- import numpy as np
4
-
5
- # Load model
6
- model = tf.keras.models.load_model("waste_classification_finetuned.h5")
7
- IMG_SIZE = (224, 224)
8
-
9
- # Class labels
10
- index_to_label = {
11
- 0: "paper", 1: "e-waste", 2: "metal", 3: "light blubs",
12
- 4: "organic", 5: "plastic", 6: "clothes", 7: "glass", 8: "batteries"
13
- }
14
-
15
- def classify_image(img):
16
- img = tf.image.resize(img, IMG_SIZE)
17
- img = tf.expand_dims(img, axis=0)
18
- img = tf.cast(img, tf.float32) / 255.0
19
- pred = model.predict(img)[0]
20
- return {index_to_label[i]: float(pred[i]) for i in range(len(pred))}
21
-
22
- demo = gr.Interface(
23
- fn=classify_image,
24
- inputs=gr.Image(type="numpy", label="Upload Waste Image"),
25
- outputs=gr.Label(num_top_classes=3),
26
- title="Smart Waste Classifier"
27
- )
28
-
29
- demo.launch()