MisterAI commited on
Commit
fe2644c
·
verified ·
1 Parent(s): 22b2e1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -9
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(model_path=model_path, verbose=True)
 
 
 
 
 
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=2000,
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(lines=10, label="Entrez votre texte"),
 
 
 
 
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()