MisterAI commited on
Commit
4d50477
·
verified ·
1 Parent(s): 710e458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -13,7 +13,7 @@ import time
13
  from PIL import Image
14
  import io
15
  import requests
16
- from diffusers import FluxPipeline
17
 
18
  # Configuration des modèles disponibles
19
  TEXT_MODELS = {
@@ -94,11 +94,11 @@ class PresentationGenerator:
94
  )
95
  print(f"Modèle Transformers {model_id} chargé avec succès!")
96
 
97
- def load_image_model(self, model_name):
98
- """Charge le modèle de génération d'images"""
99
- model_id = IMAGE_MODELS[model_name]
100
- self.image_pipeline = gr.load("models/black-forest-labs/FLUX.1-schnell")
101
- print(f"Modèle d'image FLUX chargé : {model_id}")
102
 
103
  def generate_text(self, prompt, temperature=0.7, max_tokens=4096):
104
  """Génère le texte de la présentation"""
 
13
  from PIL import Image
14
  import io
15
  import requests
16
+ #from diffusers import FluxPipeline
17
 
18
  # Configuration des modèles disponibles
19
  TEXT_MODELS = {
 
94
  )
95
  print(f"Modèle Transformers {model_id} chargé avec succès!")
96
 
97
+ # def load_image_model(self, model_name):
98
+ # """Charge le modèle de génération d'images"""
99
+ # model_id = IMAGE_MODELS[model_name]
100
+ # self.image_pipeline = gr.load("models/black-forest-labs/FLUX.1-schnell")
101
+ # print(f"Modèle d'image FLUX chargé : {model_id}")
102
 
103
  def generate_text(self, prompt, temperature=0.7, max_tokens=4096):
104
  """Génère le texte de la présentation"""