Update app.py

app.py

@@ -4,20 +4,10 @@ from tensorflow.keras.preprocessing import image
 import numpy as np
 from PIL import Image
 from keras import layers
-from transformers import TFAutoModelForImageClassification
-from transformers import AutoImageProcessor
 
-# Load model#'model = tf.keras.models.load_model("xception-head")
-
-# Replace with your Hugging Face model repository name
-model_name = "icputrd/Inception-V3-Human-Bodypart-Classifier"
-
-# Load the pre-trained TensorFlow model from Hugging Face
-model = TFAutoModelForImageClassification.from_pretrained(model_name)
-
-# Load the associated image processor (for preprocessing input images)
-image_processor = AutoImageProcessor.from_pretrained(model_name)
 
+# Load your trained Xception model
+model = tf.keras.models.load_model("xception-head")
 
 # Define the labels for your classification
 class_labels = ['arm', 'hand', 'foot',  'legs','fullbody','head','backside', 'torso', 'stake', 'plastic'] # Replace with your actual class names