Spaces:
Sleeping
Sleeping
aviseth commited on
Commit ·
45f4b13
1
Parent(s): c5c3134
fix: sync latest inference and api fixes
Browse files- 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(
|