Update main.py

main.py

@@ -1,6 +1,7 @@
-from fastapi import FastAPI, Response
+from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from transformers import pipeline
+from pydantic import BaseModel
 
 app = FastAPI()
 
@@ -23,6 +24,6 @@ def infer_endpoint(data: dict):
         result = t5(input_text)
         return result
 
-@app.get("/", response_class=Response)
-async def index():
-    return app.FileResponse("./static/index.html")
+@app.get('/')
+def index():
+    return app.FileResponse('./static/index.html')