opus-en-pt-api / test.py
playmak3r's picture
fix run_batch inference function
1eabbe3
import time
import requests
from typing import List
def test(text: str):
start_time = time.perf_counter()
response = requests.get(f"http://localhost:7860/api/translate?text={text}")
print(response.status_code)
print(response.content)
end_time = time.perf_counter()
print(f"Tempo de execução: {end_time - start_time:.4f} segundos")
def test_batch(texts: List[str]):
start_time = time.perf_counter()
response = requests.post("http://localhost:7860/api/translate", json={
"input_texts": texts
})
print(response.status_code)
print(response.content)
end_time = time.perf_counter()
print(f"Tempo de execução: {end_time - start_time:.4f} segundos")
if __name__ == "__main__":
test("Now let's make my mum's favourite. So three mars bars into the pan. Then we add the tuna and just stir for a bit, just let the chocolate and fish infuse. A sprinkle of olive oil and some tomato ketchup. Now smell that. Oh boy this is going to be incredible.")
test_batch([
"Tom tried to stab me.",
"He has been to Hawaii several times."
])