Update app.py

app.py

@@ -9,12 +9,12 @@ app = FastAPI()
 # Load the image classification pipeline
 pipe = pipeline("image-classification", model="mateoluksenberg/dit-base-Classifier_CM05")
 
-class Message(BaseModel):
-    text: str
-
 @app.post("/test/")
-async def test_endpoint(message: Message):
-    response = {"message": f"Received your message: {message.text}"}
+async def test_endpoint(message: dict):
+    if "text" not in message:
+        raise HTTPException(status_code=400, detail="Missing 'text' in request body")
+    
+    response = {"message": f"Received your message: {message['text']}"}
     return response
     
 @app.post("/classify/")