made tailing / optional

src/main.py

@@ -63,7 +63,7 @@ async def gpu_check():
 # Translation Task
 # =========================
 @app.post(
-    "/translation/{model_name:path}/", 
+    "/translation/{model_name:path}", 
     openapi_extra={
         "requestBody": {
             "content": {
@@ -92,6 +92,7 @@ async def translate(
         list: The translation result(s) as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     translationTaskService = TranslationTaskService(logger)
     return await translationTaskService.translate(request, model_name)
 
@@ -100,7 +101,7 @@ async def translate(
 # Zero-Shot Image Classification Task
 # =========================
 @app.post(
-    "/zero-shot-image-classification/{model_name:path}/",
+    "/zero-shot-image-classification/{model_name:path}",
     openapi_extra={
         "requestBody": {
             "content": {
@@ -129,6 +130,7 @@ async def zero_shot_image_classification(
         list: The classification result(s) as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     zeroShotTask = ClassificationTaskService(logger, 'zero-shot-image-classification')
     return await zeroShotTask.classify(request, model_name)
 
@@ -137,7 +139,7 @@ async def zero_shot_image_classification(
 # Image Classification Task
 # =========================
 @app.post(
-    "/image-classification/{model_name:path}/",
+    "/image-classification/{model_name:path}",
     openapi_extra={
         "requestBody": {
             "content": {
@@ -165,6 +167,7 @@ async def image_classification(
         list: The classification result(s) as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     imageTask = ClassificationTaskService(logger, 'image-classification')
     return await imageTask.classify(request, model_name)
 
@@ -174,7 +177,7 @@ async def image_classification(
 # Zero-Shot Text Classification Task
 # =========================
 @app.post(
-    "/zero-shot-text-classification/{model_name:path}/",
+    "/zero-shot-text-classification/{model_name:path}",
     openapi_extra={
         "requestBody": {
             "content": {
@@ -203,6 +206,7 @@ async def zero_shot_text_classification(
         list: The classification result(s) as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     zeroShotTask = ClassificationTaskService(logger, 'zero-shot-classification')
     return await zeroShotTask.classify(request, model_name)
 
@@ -211,7 +215,7 @@ async def zero_shot_text_classification(
 # Text Classification Task
 # =========================
 @app.post(
-    "/text-classification/{model_name:path}/",
+    "/text-classification/{model_name:path}",
     openapi_extra={
         "requestBody": {
             "content": {
@@ -239,6 +243,7 @@ async def text_classification(
         list: The classification result(s) as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     textTask = ClassificationTaskService(logger, 'text-classification')
     return await textTask.classify(request, model_name)
 
@@ -250,7 +255,7 @@ async def text_classification(
 # Image to Text Task
 # =========================
 @app.post(
-    "/image-to-text/{model_name:path}/",
+    "/image-to-text/{model_name:path}",
     openapi_extra={
         "requestBody": {
             "content": {
@@ -286,5 +291,6 @@ async def image_to_text(
         list: The generated text as returned by the pipeline.
     """
 
+    model_name = model_name.rstrip("/")
     imageToTextTask = TextToImageTaskService(logger)
     return await imageToTextTask.extract(request, model_name)