aviseth commited on
Commit
45f4b13
·
1 Parent(s): c5c3134

fix: sync latest inference and api fixes

Browse files
Files changed (1) hide show
  1. src/api/main.py +2 -0
src/api/main.py CHANGED
@@ -152,6 +152,8 @@ async def predict(request: PredictionRequest, background_tasks: BackgroundTasks)
152
  from src.models.inference import predict as run_inference
153
  result = run_inference(text, model_key)
154
  except Exception as e:
 
 
155
  raise HTTPException(status_code=500, detail=f"Inference error: {e}")
156
 
157
  response = PredictionResponse(
 
152
  from src.models.inference import predict as run_inference
153
  result = run_inference(text, model_key)
154
  except Exception as e:
155
+ import traceback
156
+ traceback.print_exc()
157
  raise HTTPException(status_code=500, detail=f"Inference error: {e}")
158
 
159
  response = PredictionResponse(