Spaces:

scott2srikanth
/

askmynotes_classifier

Sleeping

Update app.py

9700325 verified 6 months ago

486 Bytes

from fastapi import FastAPI
from pydantic import BaseModel
import joblib

app = FastAPI()

model = joblib.load("askmynotes_classifier.pkl")

class QuestionRequest(BaseModel):
    question: str

@app.get("/")
def home():
    return {"message": "AskMyNotes Classifier API Running"}

@app.post("/predict")
def predict(data: QuestionRequest):

    prediction = model.predict([data.question])[0]

    return {
        "question": data.question,
        "predicted_category": prediction
    }

	from fastapi import FastAPI
	from pydantic import BaseModel
	import joblib

	app = FastAPI()

	model = joblib.load("askmynotes_classifier.pkl")

	class QuestionRequest(BaseModel):
	question: str

	@app.get("/")
	def home():
	return {"message": "AskMyNotes Classifier API Running"}

	@app.post("/predict")
	def predict(data: QuestionRequest):

	prediction = model.predict([data.question])[0]

	return {
	"question": data.question,
	"predicted_category": prediction
	}