KeivanR commited on
Commit
a7cafdb
·
1 Parent(s): 5a63674

timeout predict 60

Browse files
Files changed (1) hide show
  1. qwen_classifier/predict.py +1 -1
qwen_classifier/predict.py CHANGED
@@ -36,7 +36,7 @@ def _predict_hf_api(text, hf_token=None):
36
  "Authorization": f"Bearer {hf_token}",
37
  "Content-Type": "application/json"
38
  } if hf_token else {"Content-Type": "application/json"},
39
- timeout=10
40
  )
41
  response.raise_for_status() # Raise HTTP errors
42
  return response.json()
 
36
  "Authorization": f"Bearer {hf_token}",
37
  "Content-Type": "application/json"
38
  } if hf_token else {"Content-Type": "application/json"},
39
+ timeout=60
40
  )
41
  response.raise_for_status() # Raise HTTP errors
42
  return response.json()