annanau commited on
Commit
9bb9c46
·
verified ·
1 Parent(s): 758c282

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("xception-head")
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 huma. 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,
 
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,