pmv-hou/linkedin_post_generator

1 2import gradio as gr3from huggingface_hub import InferenceClient4import os5 6API_TOKEN = os.environ['HF_TOKEN']7client = InferenceClient()8 9def query(payload):10    response = client.text_generation(payload, model="google/gemma-7b", max_new_tokens=12)11    return response #.json()12    13 14text_generation_interface = gr.Interface(15    fn=query,16    inputs=[17        gr.Textbox(label="Input Text"),18    ],19    outputs=gr.Textbox(label="Generated Text"),20    title="google/gemma-7b",21    22).launch()