rr

app.py

@@ -8,34 +8,43 @@ def generate_poem(author, style, syllables):
     # Construire le prompt de manière plus explicite
     prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
 
-    response = openchat_model(prompt_text)
-    print("Raw Model Response:", response)
+    # Utiliser le modèle chargé pour générer un poème
+    generated_poem = openchat_model(prompt_text)
 
-    return response["choices"][0]["message"]["content"]
+    return generated_poem
 
 # 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"),
-    ],
-    outputs="text"  # Nous renvoyons un texte généré par le modèle
+    inputs=[gr.Textbox("Author"), gr.Textbox("Style"), gr.Textbox("Syllables")],
+    outputs=gr.Textbox(),  # Nous renvoyons un texte généré par le modèle
 )
 
 # Lancer l'interface
 if __name__ == "__main__":
     demo.launch()
+import gradio as gr
 
+# Charger le modèle OpenChat en dehors de la fonction pour éviter de le charger à chaque appel
+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):
+    # Construire le prompt de manière plus explicite
+    prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
 
+    # Utiliser le modèle chargé pour générer un poème
+    generated_poem = openchat_model(prompt_text)
 
+    return generated_poem
 
+# Créer une interface Gradio avec les entrées et sorties spécifiées
+demo = gr.Interface(
+    fn=generate_poem,
+    inputs=[gr.Textbox("Author"), gr.Textbox("Style"), gr.Textbox("Syllables")],
+    outputs=gr.Textbox(),  # Nous renvoyons un texte généré par le modèle
+)
 
-""""import gradio as gr
-def get_name_with_hello(name: str) -> str:
- return 'Hello ' + name + '! '
-if __name__ == '__main__':
- demo = gr.Interface(fn=get_name_with_hello, inputs='text', outputs='text')
- demo.launch()"""
+# Lancer l'interface
+if __name__ == "__main__":
+    demo.launch()

flagged/log.csv

@@ -0,0 +1,2 @@
+author,style,syllables,output,flag,username,timestamp
+Baudelaire,Moderne,12,,,,2023-12-21 20:35:54.873318

hello.py

@@ -0,0 +1,9 @@
+import gradio as gr
+
+def greet(name):
+    return "Hello " + name + "!"
+
+demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+    
+if __name__ == "__main__":
+    demo.launch(show_api=False)