diff --git a/myproject/docker-compose.yml b/myproject/docker-compose.yml
index f7530257616880e6d6b942cdc11bea8572e5829a..cbfb0f2593a106dc407dfbdb3693a716d18d82c9 100644
--- a/myproject/docker-compose.yml
+++ b/myproject/docker-compose.yml
@@ -41,8 +41,6 @@ services:
 
   tensorflow_serving:
     image: tensorflow/serving
-    ports:
-      - "8501:8501"
     volumes:
       - ./models:/models
     environment:
diff --git a/myproject/myapp/views.py b/myproject/myapp/views.py
index 40b209f8e69fe66fa21e32fe37ccc974471ac001..b99c876f7580529dcee0a6a5c7c4595140af8c31 100644
--- a/myproject/myapp/views.py
+++ b/myproject/myapp/views.py
@@ -262,6 +262,17 @@ def generate_pdf(request):
 # Running the audio file through the model
 class InstrumentDetectionView(APIView):
     def post(self, request):
+        # Get the user's token count
+        user_token_count = UserTokenCount.objects.get(user=request.user)
+
+        # Check if the user has more than one token
+        if user_token_count.token_count < 1:
+            return Response({'error': 'Insufficient tokens'}, status=status.HTTP_403_FORBIDDEN)
+
+        # Decrease the user's token count by one
+        user_token_count.token_count -= 1
+        user_token_count.save()
+        
         serializer = InstrumentDetectionSerializer(data=request.data)
         if serializer.is_valid():
             audio_file = serializer.validated_data['audio_file']