Update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,13 @@ model_path = hf_hub_download(
|
|
37 |
repo_type="model"
|
38 |
)
|
39 |
|
40 |
-
# Initialisation du modèle
|
41 |
-
text_to_presentation = Llama(
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
def parse_presentation_content(content):
|
44 |
"""Parse le contenu généré en sections pour les diapositives"""
|
@@ -93,7 +98,7 @@ def generate_presentation(text):
|
|
93 |
# Génération du contenu avec le modèle
|
94 |
response = text_to_presentation(
|
95 |
full_prompt,
|
96 |
-
max_tokens=
|
97 |
temperature=0.7,
|
98 |
stop=["<end>"],
|
99 |
echo=False
|
@@ -112,17 +117,18 @@ def generate_presentation(text):
|
|
112 |
|
113 |
return f"Présentation générée avec succès ! Vous pouvez la télécharger ici : {os.path.abspath(output_path)}"
|
114 |
|
115 |
-
# Interface Gradio
|
116 |
demo = gr.Interface(
|
117 |
fn=generate_presentation,
|
118 |
-
inputs=gr.Textbox(
|
|
|
|
|
|
|
|
|
119 |
outputs=gr.Textbox(label="Statut"),
|
120 |
title="Générateur de Présentations PowerPoint",
|
121 |
description="Entrez votre texte et obtenez une présentation PowerPoint générée automatiquement."
|
122 |
)
|
123 |
|
124 |
if __name__ == "__main__":
|
125 |
-
demo.launch()
|
126 |
-
|
127 |
-
|
128 |
-
|
|
|
37 |
repo_type="model"
|
38 |
)
|
39 |
|
40 |
+
# Initialisation du modèle avec des paramètres de contexte plus grands
|
41 |
+
text_to_presentation = Llama(
|
42 |
+
model_path=model_path,
|
43 |
+
verbose=True,
|
44 |
+
n_ctx=4096, # Taille maximale du contexte (entrée + sortie)
|
45 |
+
n_batch=256 # Taille du batch pour le traitement
|
46 |
+
)
|
47 |
|
48 |
def parse_presentation_content(content):
|
49 |
"""Parse le contenu généré en sections pour les diapositives"""
|
|
|
98 |
# Génération du contenu avec le modèle
|
99 |
response = text_to_presentation(
|
100 |
full_prompt,
|
101 |
+
max_tokens=4096, # Nombre maximum de tokens en sortie
|
102 |
temperature=0.7,
|
103 |
stop=["<end>"],
|
104 |
echo=False
|
|
|
117 |
|
118 |
return f"Présentation générée avec succès ! Vous pouvez la télécharger ici : {os.path.abspath(output_path)}"
|
119 |
|
120 |
+
# Interface Gradio avec une zone de texte plus grande
|
121 |
demo = gr.Interface(
|
122 |
fn=generate_presentation,
|
123 |
+
inputs=gr.Textbox(
|
124 |
+
lines=10,
|
125 |
+
label="Entrez votre texte",
|
126 |
+
max_lines=50 # Permet plus de lignes de texte
|
127 |
+
),
|
128 |
outputs=gr.Textbox(label="Statut"),
|
129 |
title="Générateur de Présentations PowerPoint",
|
130 |
description="Entrez votre texte et obtenez une présentation PowerPoint générée automatiquement."
|
131 |
)
|
132 |
|
133 |
if __name__ == "__main__":
|
134 |
+
demo.launch()
|
|
|
|
|
|