Update app.py

app.py

@@ -17,6 +17,18 @@ if not hf_token:
 # Initialize the Hugging Face API
 api = HfApi()
 
+# Repository Details
+REPO_NAME = "face_and_emotion_detection"  # Replace with your desired repo name
+REPO_ID = "LovnishVerma/" + REPO_NAME  # Update "LovnishVerma" with your Hugging Face username
+REPO_TYPE = "space"  # Set to 'space' for Streamlit projects
+
+# Ensure repository exists
+try:
+    api.create_repo(repo_id=REPO_ID, repo_type=REPO_TYPE, token=hf_token, exist_ok=True)
+    st.success(f"Repository '{REPO_NAME}' is ready on Hugging Face!")
+except Exception as e:
+    st.error(f"Error creating repository: {e}")
+
 # Database setup
 DATABASE = "students.db"
 
@@ -57,12 +69,11 @@ def save_image_to_hugging_face(image, roll_no):
 
     # Upload to Hugging Face
     try:
-        repo_name = "LovnishVerma/face_and_emotion_detection"  # Replace with your repo details
         api.upload_file(
             path_or_fileobj=local_path,
             path_in_repo=filename,
-            repo_id=repo_name,
-            repo_type="dataset",
+            repo_id=REPO_ID,
+            repo_type=REPO_TYPE,
             token=hf_token  # Pass the token directly
         )
         st.success(f"Image uploaded to Hugging Face: {filename}")