Spaces:

thisisgini
/

ecommerce_bot

Runtime error

Update app.py

aa29669 verified 7 months ago

741 Bytes

from fastapi import FastAPI, Request
from inference import run_inference  # Importing from your root inference.py
import uvicorn

app = FastAPI()

# Health Check Endpoint
@app.get("/")
async def health_check():
    return {"status": "healthy", "message": "Ecommerce Bot API is live"}

# Prediction Endpoint
@app.post("/predict")
async def predict(request: Request):
    data = await request.json()
    # Assuming your inference.py has a function called run_inference
    result = run_inference(data) 
    return {"prediction": result}

# The Reset Endpoint (which your logs show is already working)
@app.post("/reset")
async def reset():
    return {"status": "ok"}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)

	from fastapi import FastAPI, Request
	from inference import run_inference # Importing from your root inference.py
	import uvicorn

	app = FastAPI()

	# Health Check Endpoint
	@app.get("/")
	async def health_check():
	return {"status": "healthy", "message": "Ecommerce Bot API is live"}

	# Prediction Endpoint
	@app.post("/predict")
	async def predict(request: Request):
	data = await request.json()
	# Assuming your inference.py has a function called run_inference
	result = run_inference(data)
	return {"prediction": result}

	# The Reset Endpoint (which your logs show is already working)
	@app.post("/reset")
	async def reset():
	return {"status": "ok"}

	if __name__ == "__main__":
	uvicorn.run(app, host="0.0.0.0", port=7860)