Change envs

.github/workflows/main.yml

@@ -35,8 +35,9 @@ jobs:
         run: |
           python3 inference.py
   
-  DeployDev:
-    name: Deploy to Development
+  DeployStaging:
+    name: Deploy to Staging
+    if: github.event.ref == 'refs/head/main'
     needs: [Build]
     runs-on: ubuntu-latest
     environment:
@@ -57,8 +58,11 @@ jobs:
 
   DeployProd:
     name: Deploy to Production 
-    needs: [DeployDev]
+    needs: [DeployStaging]
     runs-on: ubuntu-latest
+    environment:
+      name: Production
+      url: 'https://huggingface.co/spaces/cccmatthew/gp30-deployment'
     steps:
       - uses: actions/checkout@v2
         with: