Spaces:
Sleeping
Sleeping
Camille Lebrun
commited on
Commit
•
ff948b9
1
Parent(s):
08be1ea
rr
Browse files- app.py +24 -15
- flagged/log.csv +2 -0
- hello.py +9 -0
app.py
CHANGED
@@ -8,34 +8,43 @@ def generate_poem(author, style, syllables):
|
|
8 |
# Construire le prompt de manière plus explicite
|
9 |
prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
|
10 |
|
11 |
-
|
12 |
-
|
13 |
|
14 |
-
return
|
15 |
|
16 |
# Créer une interface Gradio avec les entrées et sorties spécifiées
|
17 |
demo = gr.Interface(
|
18 |
fn=generate_poem,
|
19 |
-
inputs=[
|
20 |
-
|
21 |
-
gr.Radio(["Classique", "Moderne"], label="Choisir le style du poème"),
|
22 |
-
gr.Radio(["alexandrin", "octosyllabe", "décasyllabe"], label="Choisir le nombre de syllabes"),
|
23 |
-
],
|
24 |
-
outputs="text" # Nous renvoyons un texte généré par le modèle
|
25 |
)
|
26 |
|
27 |
# Lancer l'interface
|
28 |
if __name__ == "__main__":
|
29 |
demo.launch()
|
|
|
30 |
|
|
|
|
|
31 |
|
|
|
|
|
|
|
|
|
32 |
|
|
|
|
|
33 |
|
|
|
34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
if __name__ == '__main__':
|
40 |
-
demo = gr.Interface(fn=get_name_with_hello, inputs='text', outputs='text')
|
41 |
-
demo.launch()"""
|
|
|
8 |
# Construire le prompt de manière plus explicite
|
9 |
prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
|
10 |
|
11 |
+
# Utiliser le modèle chargé pour générer un poème
|
12 |
+
generated_poem = openchat_model(prompt_text)
|
13 |
|
14 |
+
return generated_poem
|
15 |
|
16 |
# Créer une interface Gradio avec les entrées et sorties spécifiées
|
17 |
demo = gr.Interface(
|
18 |
fn=generate_poem,
|
19 |
+
inputs=[gr.Textbox("Author"), gr.Textbox("Style"), gr.Textbox("Syllables")],
|
20 |
+
outputs=gr.Textbox(), # Nous renvoyons un texte généré par le modèle
|
|
|
|
|
|
|
|
|
21 |
)
|
22 |
|
23 |
# Lancer l'interface
|
24 |
if __name__ == "__main__":
|
25 |
demo.launch()
|
26 |
+
import gradio as gr
|
27 |
|
28 |
+
# Charger le modèle OpenChat en dehors de la fonction pour éviter de le charger à chaque appel
|
29 |
+
openchat_model = gr.load("models/openchat/openchat_3.5")
|
30 |
|
31 |
+
# Définir une fonction de génération de poème en fonction des entrées
|
32 |
+
def generate_poem(author, style, syllables):
|
33 |
+
# Construire le prompt de manière plus explicite
|
34 |
+
prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
|
35 |
|
36 |
+
# Utiliser le modèle chargé pour générer un poème
|
37 |
+
generated_poem = openchat_model(prompt_text)
|
38 |
|
39 |
+
return generated_poem
|
40 |
|
41 |
+
# Créer une interface Gradio avec les entrées et sorties spécifiées
|
42 |
+
demo = gr.Interface(
|
43 |
+
fn=generate_poem,
|
44 |
+
inputs=[gr.Textbox("Author"), gr.Textbox("Style"), gr.Textbox("Syllables")],
|
45 |
+
outputs=gr.Textbox(), # Nous renvoyons un texte généré par le modèle
|
46 |
+
)
|
47 |
|
48 |
+
# Lancer l'interface
|
49 |
+
if __name__ == "__main__":
|
50 |
+
demo.launch()
|
|
|
|
|
|
flagged/log.csv
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
author,style,syllables,output,flag,username,timestamp
|
2 |
+
Baudelaire,Moderne,12,,,,2023-12-21 20:35:54.873318
|
hello.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
|
3 |
+
def greet(name):
|
4 |
+
return "Hello " + name + "!"
|
5 |
+
|
6 |
+
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
|
7 |
+
|
8 |
+
if __name__ == "__main__":
|
9 |
+
demo.launch(show_api=False)
|