Update main.py

main.py

@@ -3,7 +3,8 @@ from pydantic import BaseModel
 from typing import List
 
 class Item(BaseModel):
-    imageArray: List[bytes]
+    name: str
+    image1: bytes
 
 app = FastAPI()
 
@@ -11,6 +12,10 @@ app = FastAPI()
 def read_root():
     return {"Hello": "World!"}
 
+@app.put("/return/")
+def return_item(item: Item):
+   return item
+
 @app.put("/")
 def predict(Item):
     # Load the OpenAI CLIP Model