Simplify manage to autodetect task+framework if possible. (#122)

* Simplify manage to autodetect task+framework if possible.

* Style.

* Adding huggingface_hub as test dependency.

* Actions should run on PRs not on push.

* Fix failing test.

* Alphabetical order.

* Using library_name

.github/workflows/python-api-allennlp.yaml

@@ -1,7 +1,7 @@
 name: allennlp-docker
 
 on:
-  push:
+  pull-request:
     paths:
       - "api-inference-community/docker_images/allennlp/**"
 jobs:

.github/workflows/python-api-quality.yaml

@@ -1,7 +1,7 @@
 name: Inference API code quality
 
 on:
-  push:
+  pull_request:
     paths:
       - "api-inference-community/**"
 

.github/workflows/python-api-tests.yaml

@@ -1,6 +1,6 @@
 name: Inference API Python-tests
 on:
-  push:
+  pull_request:
     paths:
       - "api-inference-community/**"
 jobs:
@@ -25,7 +25,7 @@ jobs:
         working-directory: api-inference-community
         run: |
           pip install --upgrade pip
-          pip install pytest pillow httpx
+          pip install pytest pillow httpx huggingface_hub
           pip install -e .
       - run: make test
         working-directory: api-inference-community

api-inference-community/api_inference_community/validation.py

@@ -172,6 +172,7 @@ class StringInput(BaseModel):
     "token-classification": StringInput,
     "translation": StringInput,
     "zero-shot-classification": StringInput,
+    "text-to-speech": StringInput,
 }
 
 BATCH_ENABLED_PIPELINES = ["feature-extraction"]

api-inference-community/manage.py

@@ -5,6 +5,8 @@
 import subprocess
 import uuid
 
+from huggingface_hub import HfApi
+
 
 class cd:
     """Context manager for changing the current working directory"""
@@ -63,6 +65,13 @@ def show(args):
                         print(" " * 4, key.value)
 
 
+def resolve(model_id: str) -> [str, str]:
+    info = HfApi().model_info(model_id)
+    task = info.pipeline_tag
+    framework = info.library_name
+    return task, framework.replace("-", "_")
+
+
 def start(args):
     import sys
 
@@ -71,6 +80,14 @@ def start(args):
     model_id = args.model_id
     task = args.task
     framework = args.framework
+    if task is None or framework is None:
+        rtask, rframework = resolve(model_id)
+        if task is None:
+            task = rtask
+            print(f"Inferred task : {task}")
+        if framework is None:
+            framework = rframework
+            print(f"Inferred framework : {framework}")
 
     local_path = os.path.join(
         os.path.dirname(os.path.dirname(__file__)), "docker_images", framework
@@ -85,6 +102,12 @@ def docker(args):
     model_id = args.model_id
     task = args.task
     framework = args.framework
+    if task is None or framework is None:
+        rtask, rframework = resolve(model_id)
+        if task is None:
+            task = rtask
+        if framework is None:
+            framework = rframework
 
     tag = create_docker(framework)
     run_docker_command = [
@@ -124,13 +147,11 @@ def main():
     parser_start.add_argument(
         "--task",
         type=str,
-        required=True,
         help="Which task to load",
     )
     parser_start.add_argument(
         "--framework",
         type=str,
-        required=True,
         help="Which framework to load",
     )
     parser_start.set_defaults(func=start)
@@ -146,13 +167,11 @@ def main():
     parser_docker.add_argument(
         "--task",
         type=str,
-        required=True,
         help="Which task to load",
     )
     parser_docker.add_argument(
         "--framework",
         type=str,
-        required=True,
         help="Which framework to load",
     )
     parser_docker.set_defaults(func=docker)

api-inference-community/tests/test_routes.py

@@ -124,7 +124,7 @@ async def startup_event():
             app.get_pipeline = get_pipeline
 
         with TestClient(app) as client:
-            response = client.post("/", data=b"")
+            response = client.post("/", data=b"2222")
 
         self.assertEqual(
             response.status_code,