Spaces:
Sleeping
Sleeping
Download src/predict.py from Vivekkrishu/chatbot: direct link, hf CLI and curl.
- Browser
- Download file 387 Bytes
-
https://huggingface.co/spaces/Vivekkrishu/chatbot/resolve/main/src/predict.py
- Command line
-
hf download hf://spaces/Vivekkrishu/chatbot/src/predict.py
-
curl -L -o predict.py https://huggingface.co/spaces/Vivekkrishu/chatbot/resolve/main/src/predict.py
387 Bytes
| import joblib | |
| from preprocess import clean_text | |
| import random | |
| # Load model & responses | |
| model = joblib.load("models/lms_chatbot.joblib") | |
| responses = joblib.load("models/responses.joblib") | |
| def chatbot_response(user_input: str) -> str: | |
| cleaned = clean_text(user_input) | |
| tag = model.predict([cleaned])[0] | |
| return random.choice(responses.get(tag, ["Sorry, I don't understand."])) | |