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()
|
|
|
|
|
|
|
|
|