Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from huggingface_hub import InferenceClient | |
| themes=["Pick a Genre", "thriller", "fantasy", "romance", "sci-fi", "mystery", "fiction"] | |
| def reaction(text, theme): | |
| modelName= 'HuggingFaceH4/zephyr-7b-beta' | |
| thiscontent=f'Give only one {theme} response to this statement: {text}?' | |
| messages=[{'role': 'user', 'content': thiscontent}] | |
| client=InferenceClient(model=modelName) | |
| output=client.chat_completion(messages, max_tokens=100) | |
| reaction=output.choices[0].message.content | |
| return reaction | |
| with gr.Blocks(theme=gr.themes.Citrus()) as demo: | |
| with gr.Row(): | |
| with gr.Column(): | |
| text = gr.Textbox(label="What\'s your idea",scale=1) | |
| theme=gr.Dropdown(themes, label="genre") | |
| react_btn = gr.Button("Generate",scale=1) | |
| output = gr.Textbox(label="Reaction",scale=1) | |
| with gr.Column(): | |
| react_btn.click(fn=reaction, inputs=[text, theme], outputs=output) | |
| demo.launch(debug=True, share=True) |