chatbot_model.py

from transformers import pipeline

# Load a simple model for chatbot generation
chatbot = pipeline("text-generation", model="facebook/blenderbot-400M-distill")

def get_chatbot_response(prompt):
    response = chatbot(prompt, max_length=120, num_return_sequences=1)
    return response[0]['generated_text']