Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from keras import layers
|
|
7 |
|
8 |
|
9 |
# Load your trained Xception model
|
10 |
-
model = tf.keras.models.load_model("
|
11 |
|
12 |
# Define the labels for your classification
|
13 |
class_labels = ['arm', 'hand', 'foot', 'legs','fullbody','head','backside', 'torso', 'stake', 'plastic'] # Replace with your actual class names
|
@@ -33,7 +33,7 @@ def classify_image(img):
|
|
33 |
demo = gr.Interface(
|
34 |
fn=classify_image,
|
35 |
title="Human Bodypart Image Classification",
|
36 |
-
description = "Predict the bodypart of
|
37 |
inputs=gr.Image(type="pil"),
|
38 |
outputs=[gr.Label(num_top_classes=len(class_labels)), gr.Number()],
|
39 |
live=True,
|
|
|
7 |
|
8 |
|
9 |
# Load your trained Xception model
|
10 |
+
model = tf.keras.models.load_model("inception_acc_0.989001-_val_acc_0.98252.h5")
|
11 |
|
12 |
# Define the labels for your classification
|
13 |
class_labels = ['arm', 'hand', 'foot', 'legs','fullbody','head','backside', 'torso', 'stake', 'plastic'] # Replace with your actual class names
|
|
|
33 |
demo = gr.Interface(
|
34 |
fn=classify_image,
|
35 |
title="Human Bodypart Image Classification",
|
36 |
+
description = "Predict the bodypart of human bodypart images. This is a demo of our human bodypart image <a href=\"https://huggingface.co/icputrd/Inception-V3-Human-Bodypart-Classifier\">classifier</a>.",
|
37 |
inputs=gr.Image(type="pil"),
|
38 |
outputs=[gr.Label(num_top_classes=len(class_labels)), gr.Number()],
|
39 |
live=True,
|