Update app.py

app.py

@@ -22,7 +22,7 @@ class ImageRequest(BaseModel):  # Новая модель для запроса
     is_public: bool = False
 
 @app.post("/generate-image")
-async def generate_image(request: ImageRequest):
+async def generate_image(request: GenerationRequest):
 
 
 
@@ -30,7 +30,7 @@ async def generate_image(request: ImageRequest):
     try:
         pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")
         
-        prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+        prompt = request.prompt
         image = pipe(prompt).images[0]        
         
         return image