test

app.py

@@ -0,0 +1,34 @@
+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()