Spaces:
Sleeping
Sleeping
File size: 1,312 Bytes
7b1dbbb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from click import prompt
import gradio as gr
# Charger le modèle OpenChat
openchat_model = gr.load("models/openchat/openchat_3.5")
# Définir une fonction de génération de poème en fonction des entrées
def generate_poem(author, style, syllables, activities):
# Convertir la chaîne d'activités en liste
activities_list = activities.split(", ") if activities else []
# Construire le prompt
prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}, Activities: {activities_list}"
# Appeler le modèle OpenChat avec le prompt
response = openchat_model(prompt_text)
return response["choices"][0]["message"]["content"]
# Créer une interface Gradio avec les entrées et sorties spécifiées
demo = gr.Interface(
fn=generate_poem,
inputs=[
gr.Dropdown(["Baudelaire", "La Fontaine"], label="Choisir le style d'auteur"),
gr.Radio(["Classique", "Moderne"], label="Choisir le style du poème"),
gr.Radio(["alexandrin", "octosyllabe", "décasyllabe"], label="Choisir le nombre de syllabes"),
gr.Textbox("Activités (séparées par des virgules)", label="Ajouter des activités"),
],
outputs="text" # Nous renvoyons un texte généré par le modèle
)
# Lancer l'interface
if __name__ == "__main__":
demo.launch()
|