Adding check to only operate on main

app.py

@@ -72,6 +72,12 @@ app = WebhooksServer(ui=ui.queue(), webhook_secret=WEBHOOK_SECRET)
 async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
     if not payload.event.scope.startswith("repo"):
         return Response("No task scheduled", status_code=status.HTTP_200_OK)
+    # Only run if change is on main branch
+    try:
+        if payload.updatedRefs[0].ref != 'refs/heads/main':
+            return Response("No task scheduled", status_code=status.HTTP_200_OK)
+    except:
+        return Response("No task scheduled", status_code=status.HTTP_200_OK)
 
     logger.info(f"Webhook received from {payload.repo.name} indicating a repo {payload.event.action}")
     task_queue.add_task(_process_webhook, payload=payload)

requirements.txt

@@ -1,6 +1,6 @@
 aiohttp==3.8.3
 datasets==2.18.0
-huggingface-hub>=0.20.0
+huggingface-hub==0.23.1
 pandas==1.5.3
 requests==2.31.0
 tqdm==4.66.1