MisterAI commited on
Commit
150c209
·
verified ·
1 Parent(s): c2d7e85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -99,11 +99,7 @@ class PresentationGenerator:
99
  )
100
  generated_text = response['choices'][0]['text']
101
  else:
102
- inputs = self.text_tokenizer.apply_chat_template(
103
- [{"role": "user", "content": prompt}],
104
- return_tensors="pt",
105
- return_dict=True
106
- )
107
  outputs = self.text_model.generate(
108
  **inputs,
109
  max_new_tokens=max_tokens,
@@ -182,7 +178,7 @@ def generate_skeleton(text, text_model_name, temperature, max_tokens):
182
  status = f"Squelette généré avec succès en {execution_time:.2f} secondes!"
183
 
184
  # Retourne le statut et le contenu généré
185
- return status, generated_content, None
186
 
187
  except Exception as e:
188
  print(f"Erreur lors de la génération: {str(e)}")
@@ -266,7 +262,7 @@ with gr.Blocks(theme=gr.themes.Glass()) as demo:
266
  lines=10,
267
  show_copy_button=True
268
  )
269
- create_presentation_btn = gr.Button("Créer Présentation")
270
  output_file = gr.File(
271
  label="Présentation PowerPoint"
272
  )
@@ -297,3 +293,8 @@ if __name__ == "__main__":
297
  demo.launch()
298
 
299
 
 
 
 
 
 
 
99
  )
100
  generated_text = response['choices'][0]['text']
101
  else:
102
+ inputs = self.text_tokenizer(prompt, return_tensors="pt")
 
 
 
 
103
  outputs = self.text_model.generate(
104
  **inputs,
105
  max_new_tokens=max_tokens,
 
178
  status = f"Squelette généré avec succès en {execution_time:.2f} secondes!"
179
 
180
  # Retourne le statut et le contenu généré
181
+ return status, generated_content, gr.update(visible=True)
182
 
183
  except Exception as e:
184
  print(f"Erreur lors de la génération: {str(e)}")
 
262
  lines=10,
263
  show_copy_button=True
264
  )
265
+ create_presentation_btn = gr.Button("Créer Présentation", visible=False)
266
  output_file = gr.File(
267
  label="Présentation PowerPoint"
268
  )
 
293
  demo.launch()
294
 
295
 
296
+
297
+
298
+
299
+
300
+