davanstrien HF staff commited on
Commit
6d286c3
1 Parent(s): fd21f2f

Update FastAPI import to include Query

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,7 +8,7 @@ from typing import Annotated, Any, Iterator, Union
8
  import fasttext
9
  from cashews import cache
10
  from dotenv import load_dotenv
11
- from fastapi import FastAPI, Path
12
  from httpx import AsyncClient, Client, Timeout
13
  from huggingface_hub import hf_hub_download
14
  from iso639 import Lang
@@ -248,7 +248,7 @@ async def predict_language(
248
  config: str | None = None,
249
  split: str | None = None,
250
  max_request_calls: Annotated[
251
- int, Path(title="Max number of requests to datasets server", gt=0, le=20)
252
  ] = 10,
253
  number_of_rows: int = 1000,
254
  ) -> dict[Any, Any] | None:
 
8
  import fasttext
9
  from cashews import cache
10
  from dotenv import load_dotenv
11
+ from fastapi import FastAPI, Path, Query
12
  from httpx import AsyncClient, Client, Timeout
13
  from huggingface_hub import hf_hub_download
14
  from iso639 import Lang
 
248
  config: str | None = None,
249
  split: str | None = None,
250
  max_request_calls: Annotated[
251
+ int, Query(title="Max number of requests to datasets server", gt=0, le=20)
252
  ] = 10,
253
  number_of_rows: int = 1000,
254
  ) -> dict[Any, Any] | None: