Update app.py

app.py

@@ -11,7 +11,20 @@ import json
 part_seg_model = SegformerForSemanticSegmentation.from_pretrained("Mohaddz/huggingCars")
 damage_seg_model = SegformerForSemanticSegmentation.from_pretrained("Mohaddz/DamageSeg")
 feature_extractor = AutoFeatureExtractor.from_pretrained("Mohaddz/huggingCars")
-dl_model = tf.keras.models.load_model('improved_car_damage_prediction_model.h5')
+
+# Recreate the model architecture
+def create_model(input_shape, num_classes):
+    inputs = tf.keras.Input(shape=input_shape)
+    x = tf.keras.layers.Dense(64, activation='relu')(inputs)
+    x = tf.keras.layers.Dense(32, activation='relu')(x)
+    outputs = tf.keras.layers.Dense(num_classes, activation='sigmoid')(x)
+    return tf.keras.Model(inputs=inputs, outputs=outputs)
+
+# Load model weights
+input_shape = 33  # Adjust this based on your actual input shape
+num_classes = 29  # Adjust this based on your actual number of classes
+dl_model = create_model(input_shape, num_classes)
+dl_model.load_weights('improved_car_damage_prediction_model.h5')
 
 # Load parts list
 with open('cars117.json', 'r', encoding='utf-8') as f: