Update main.py

main.py

@@ -8,27 +8,15 @@ class Airflow(BaseModel):
 
 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("/")
+@app.get("/")
 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]
-    predicted_volume = model.predict([features])[0]
-    return {"predicted_volume": predicted_volume}
+# @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]
+#     predicted_volume = model.predict([features])[0]
+#     return {"predicted_volume": predicted_volume}