Update app.py

app.py

@@ -66,6 +66,9 @@ def initialize_database():
         """)
         conn.commit()
 
+# Initialize the database
+initialize_database()
+
 def save_to_database(name, roll_no, image_path):
     """
     Saves student data (name, roll number, image path) to the SQLite database.
@@ -91,6 +94,8 @@ def save_image_to_hugging_face(image, name, roll_no):
     
     filename = f"{name}_{roll_no}.jpg"
     local_path = os.path.join(KNOWN_FACES_DIR, filename)
+    
+    # Saving the image to the correct directory
     image.save(local_path)
 
     try: