bard_model.py

from bardapi import Bard



def bard_predict(input_text, bard_api_key):

    bard = Bard(api_key=bard_api_key)
    response = bard.get_answer(input_text)
    
    return response['content']