Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from fastapi import FastAPI, HTTPException, Depends
+from fastapi import FastAPI, HTTPException, Depends, Body
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel, field_validator, ValidationError
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, GenerationConfig, StoppingCriteriaList, pipeline, StoppingCriteria
@@ -103,7 +103,7 @@ async def get_model_and_tokenizer(model_name: str):
         raise HTTPException(status_code=500, detail=f"Error loading model: {e}")
 
 @app.post("/generate")
-async def generate(request: GenerateRequest, model_resources: tuple = Depends(get_model_and_tokenizer)):
+async def generate(request: GenerateRequest = Body(...), model_resources: tuple = Depends(get_model_and_tokenizer)):
     model, tokenizer = model_resources
     try:
         model_name = request.model_name
@@ -192,7 +192,7 @@ async def load_pipeline_from_s3(task, model_name):
          raise HTTPException(status_code=500, detail=f"Error loading {task} model from S3: {e}")
 
 @app.post("/generate-image")
-async def generate_image(request: GenerateRequest):
+async def generate_image(request: GenerateRequest = Body(...)):
     try:
         if request.task_type != "text-to-image":
             raise HTTPException(status_code=400, detail="Invalid task_type for this endpoint.")
@@ -212,7 +212,7 @@ async def generate_image(request: GenerateRequest):
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
 @app.post("/generate-text-to-speech")
-async def generate_text_to_speech(request: GenerateRequest):
+async def generate_text_to_speech(request: GenerateRequest = Body(...)):
     try:
         if request.task_type != "text-to-speech":
             raise HTTPException(status_code=400, detail="Invalid task_type for this endpoint.")
@@ -232,7 +232,7 @@ async def generate_text_to_speech(request: GenerateRequest):
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
 @app.post("/generate-video")
-async def generate_video(request: GenerateRequest):
+async def generate_video(request: GenerateRequest = Body(...)):
     try:
         if request.task_type != "text-to-video":
             raise HTTPException(status_code=400, detail="Invalid task_type for this endpoint.")