Update main.py

main.py

@@ -70,6 +70,6 @@ async def predict(file: UploadFile):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during prediction: {str(e)}")
 
-# Run the application if the script is executed directly
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+@app.get("/")
+async def root():
+    return {"message": "API is up and running!"}