Create main.py

main.py

@@ -0,0 +1,27 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
+from fastapi.responses import JSONResponse
+import os
+import uvicorn
+
+app = FastAPI()
+
+# Directory where model files are stored
+MODEL_DIRECTORY = "models"
+
+@app.get("/health")
+async def api_health_check():
+    return JSONResponse(content={"status": "Service is running"})
+
+@app.get("/download/{plant_name}")
+async def download_model(plant_name: str):
+    filename = f"{plant_name}_model.keras"
+    file_path = os.path.join(MODEL_DIRECTORY, filename)
+
+    # Check if file exists
+    if not os.path.isfile(file_path):
+        raise HTTPException(status_code=404, detail=f"Model file '{filename}' not found")
+
+    return FileResponse(file_path, filename=filename, media_type="application/octet-stream")
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)