Update app.py

app.py

@@ -1,7 +1,6 @@
 ###TEST01 JUSTE CHARGER FLUX-SCHNELL
 ###
 ###
-
 import os
 import gradio as gr
 from huggingface_hub import login
@@ -46,6 +45,11 @@ class FluxGenerator:
             )
             self.pipeline.enable_model_cpu_offload()
             self.pipeline.tokenizer.add_prefix_space = False
+
+            # Forcer l'utilisation du CPU
+            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            self.pipeline.to(device)
+            print(f"Utilisation de l'appareil: {device}")
             print("Modèle FLUX chargé avec succès!")
         except Exception as e:
             print(f"Erreur lors du chargement du modèle: {str(e)}")
@@ -101,7 +105,6 @@ demo = gr.Interface(
     fn=generate,
     inputs=[
         gr.Textbox(label="Prompt", placeholder="Décrivez l'image que vous souhaitez générer..."),
-#        gr.File(label="Image ou PDF de référence (optionnel)", type="file")
         gr.File(label="Image ou PDF de référence (optionnel)", type="binary")
     ],
     outputs=gr.Image(label="Image générée"),