edit main

main.py

@@ -1,29 +1,3 @@
-<<<<<<< HEAD
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from prediction import prediction , calculate_calories
-
-# prediction_model = prediction("best.pt" , "10.jpg")
-# print(prediction_model)
-
-app = FastAPI()
-@app.get("/")
-async def read_root():
-    return {"message": "Food Model API"}
-
-@app.post("/predict")
-async def predict(file: UploadFile = File(...)):
-    if file.content_type.startswith('image'):
-        test_image = await file.read()
-        predictions = prediction("best.pt" , test_image)
-        calories = calculate_calories(predictions)
-        return {"predictions": predictions, "calories": calories}
-    else:
-        raise HTTPException(status_code=415, detail="Unsupported file type")
-    
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="localhost", port=8000)
-=======
 from fastapi import FastAPI, File, UploadFile, HTTPException , Request
 from prediction import prediction , calculate_calories
 from predictBody import predictBody
@@ -64,4 +38,3 @@ async def predict(file: UploadFile = File(...)):
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="localhost", port=8000)
->>>>>>> c87f885 (adding RFM measurement)