Update app.py

app.py

@@ -9,6 +9,14 @@ 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}"}
+    return response
+    
 @app.post("/classify/")
 async def classify_image(file: UploadFile = File(...)):
     try: