Theme_Generator / app.py
odavis1's picture
Update app.py
7562228 verified
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)