diff --git a/src/main.py b/src/main.py index 58d38b6..c7383c3 100644 --- a/src/main.py +++ b/src/main.py @@ -35,7 +35,7 @@ def model_query(prompt: str) -> str: headers=headers ) print(response.status_code) - return response.json()['text'] + return response.json()['text'][:-3].strip() def process():