Update app.py

app.py

@@ -33,8 +33,13 @@ model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.001),
               loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
               metrics=['accuracy'])
 
+
 # Using saved weights
 model.load_weights('model_weights .h5')
+class_names = {
+    1: 'Female',
+    0: 'Male'
+}
 def classify_image(image):
     # Convert Gradio Image to numpy array
     image = np.array(image)
@@ -42,9 +47,7 @@ def classify_image(image):
     image = cv2.resize(image, (180, 180))
     # Make prediction
     preds = model.predict(image[np.newaxis, ...]).squeeze()
-    y_pred = preds.argmax(axis = 0)
-    # return preds
-    # # Decode prediction
+    y_pred = preds.argmax(axis = 0) # Decode prediction
     label = class_names[int(y_pred)]
     return label
 
@@ -56,4 +59,4 @@ app = gr.Interface(
 )
 
 
-app.launch()
+app.launch(share = True)