Beaker push fix

.github/workflows/main.yml

@@ -281,14 +281,21 @@ jobs:
         env:
           BEAKER_TOKEN: ${{ secrets.BEAKER_TOKEN }}
         run: |
-          # Get the version without 'v' prefix
+          # Get version
-          VERSION=${GITHUB_REF#refs/tags/v}
+          VERSION=${{ steps.meta.outputs.version }}
 
-          # Push the Docker image to Beaker
+          # Get the image ID for the versioned tag
-          beaker image create \
+          IMAGE_ID=$(docker images --format "{{.Repository}}:{{.Tag}} {{.ID}}" | grep "alleninstituteforai/olmocr:$VERSION" | awk '{print $2}')
+
+          if [ -z "$IMAGE_ID" ]; then
+            echo "ERROR: Could not find image ID for tag alleninstituteforai/olmocr:$VERSION"
+            exit 1
+          fi
+
+          # Push the Docker image to Beaker using the image ID
+          beaker image create "$IMAGE_ID" \
             --name "olmocr-inference-$VERSION" \
-            --workspace ai2/olmocr \
+            --workspace ai2/olmocr
-            "docker://${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:$VERSION"
         
       - name: Clean up after build
         if: always()