Update app.py

app.py

@@ -17,16 +17,9 @@ app = FastAPI(lifespan=lifespan)
 
 class ImageRequest(BaseModel):  # Новая модель для запроса
     prompt: str = "girl in red dress"
-    model: str = "flux_1_schnell"
-    size: str = "1_1_HD"
-    is_public: bool = False
 
 @app.post("/generate-image")
-async def generate_image(request: GenerationRequest):
-
-
-
-    
+async def generate_image(request: ImageRequest):    
     try:
         pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")