Update app.py

app.py

@@ -77,9 +77,14 @@ def save_image_to_hugging_face(image, name, roll_no):
     """ Saves the image locally to the KNOWN_FACES_DIR and uploads it to Hugging Face. """
     filename = f"{name}_{roll_no}.jpg"
     local_path = os.path.join(KNOWN_FACES_DIR, filename)  # Ensure it's saved in the known_faces directory
-    image.save(local_path)
     
     try:
+        # Convert image to RGB if not already in RGB mode (PIL handling)
+        if image.mode != "RGB":
+            image = image.convert("RGB")
+        image.save(local_path)  # Save image locally to known_faces directory
+        
+        # Upload the file to Hugging Face
         api.upload_file(path_or_fileobj=local_path, path_in_repo=filename, repo_id=REPO_ID, repo_type="space", token=hf_token)
         st.success(f"Image uploaded to Hugging Face: {filename}")
     except Exception as e: