gleisonnanet commited on
Commit
d38295f
·
1 Parent(s): d6fcf5e

fix response

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -44,7 +44,10 @@ async def translate(request: TranslationRequest):
44
  generated_tokens, skip_special_tokens=True
45
  )[0]
46
  time_end = time.time()
47
- response = {"translation": translated_text, "computation_time": round((time_end - time_start), 3)}
 
 
 
48
  return response
49
 
50
 
 
44
  generated_tokens, skip_special_tokens=True
45
  )[0]
46
  time_end = time.time()
47
+ try:
48
+ response = {"translation": translated_text}
49
+ except Exception as E:
50
+ return {"error": str(E)}
51
  return response
52
 
53