Update main.py

main.py

@@ -84,9 +84,6 @@ transform, ort_session = initialize_model(model_file=model_file)
 class ImageRequest(BaseModel):
     image: str  # base64 encoded image
 
-# Pydantic model for response
-class TextResponse(BaseModel):
-    text: str
 
 @app.post("/predict", response_model=TextResponse)
 async def predict_captcha(request: ImageRequest):
@@ -98,10 +95,15 @@ async def predict_captcha(request: ImageRequest):
         # Get prediction
         text = get_text(img)
         
-        return TextResponse(text=text)
-    
+        return {
+            "success": True,
+            "text": text
+        }
     except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Error processing image: {str(e)}")
+        return {
+            "success": False,
+            "error": str(e)
+        }
 
 @app.get("/health")
 async def health_check():