Add application file_5

main.py

@@ -1,14 +1,32 @@
 from fastapi import FastAPI, Response, status, File, Form, UploadFile
 from pydantic import BaseModel
 import joblib
+from fastapi.middleware.cors import CORSMiddleware
 
 class Airflow(BaseModel):
     topic: str
 
 app = FastAPI()
 
+
+origins = [    "https://huggingface.co",    "https://*.huggingface.co",]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 model = joblib.load("gradient_boosting.joblib")
 
+
+@app.post("/")
+def read_root():
+    return {"Hello": "World!"}
+
+
 @app.post("/api/volume_prediction")
 async def predict_volume(vol_moving_avg: float = Form(), adj_close_rolling_med: float = Form()):
     features = [vol_moving_avg, adj_close_rolling_med]