Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Query
 import src.Paraphrase as Paraphrase
 import src.Translate as Translate
 from typing import Optional
@@ -56,9 +56,11 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
 
 @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
 # def bergamot(input_text: str, sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
-def bergamot(input_text: Union[str, list[str]], sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
+def bergamot(input_text: Union[str, list[str]] = Query(..., description="One or more input strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
     try:
         import bergamot
+        # if isinstance(input_text, str):
+        #     input_text = [input_text]
         config = bergamot.ServiceConfig(numWorkers=4)
         service = bergamot.Service(config)
         models = service.modelFromConfigPath("./deen/bergamot.config.yml")