Update app/schemas/image.py

app/schemas/image.py

@@ -26,10 +26,7 @@ class VirtualTryOnResponse(BaseModel):
     summary: Dict[str, Any]
 
 class ShoeGenerateRequest(BaseModel):
-    style: str
-    color: Optional[str] = None
-    perspective: Optional[str] = None
-    variations: Optional[int] = 1
+    prompt: str = Field(..., description="Full description of the shoe to generate")
 
 class ShoeGenerateResponse(BaseModel):
     text: Optional[str] = None