Revert "Update main.py"

app/main.py

@@ -4,8 +4,7 @@ import pathlib
 import time
 import re
 from typing import List
-from pydantic import BaseModel
-from fastapi import FastAPI
+from fastapi import FastAPI, Request, Depends
 from fastapi.middleware import Middleware
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
@@ -54,12 +53,12 @@ async def process_input(text: str):
         generator = re.split(r'(\s)', message)
     return StreamingResponse(astreamer(generator), media_type='text/event-stream')
 
-class File(BaseModel):
-    bytes: bytes
-    filename: str
+async def parse_body(request: Request):
+    data = await request.json()
+    return data
 
 @app.post("/upload")
-def upload(data: File):
+def upload(data = Depends(parse_body)):
     if data.get('bytes', None) and data.get('filename', None):
         try:
             filename = data['filename']