Update app.py

app.py

@@ -32,7 +32,7 @@ if torch.cuda.is_available():
 else:
     print("GPU não disponível. Usando CPU.")
 
-# 4. Download de Modelos
+# 5. Download de Modelos
 def download_models():
     models = [
         ("black-forest-labs/FLUX.1-Redux-dev", "flux1-redux-dev.safetensors", "style_models"),
@@ -47,11 +47,11 @@ def download_models():
     for repo_id, filename, model_type in models:
         model_dir = os.path.join(BASE_DIR, "models", model_type)
         os.makedirs(model_dir, exist_ok=True)
-        print(f"Downloading {filename} from {repo_id}...")
+        print(f"Baixando {filename} de {repo_id}...")
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=model_dir)
         folder_paths.add_model_folder_path(model_type, model_dir)
 
-# 5. Load custom nodes
+# 6. Load custom nodes
 def import_custom_nodes():
     import asyncio
     import execution
@@ -65,19 +65,22 @@ def import_custom_nodes():
     execution.PromptQueue(server_instance)
     init_extra_nodes()
 
-# 6. Main function to execute the workflow and generate an image
+# 7. Main function to execute the workflow and generate an image
 def generate_image(prompt, input_image, lora_weight, guidance, downsampling_factor, weight, seed, width, height, batch_size, steps):
     import_custom_nodes()
     
     try:
         with torch.inference_mode():
+            device = "cuda" if torch.cuda.is_available() else "cpu"
+            print(f"Using device: {device}")
+
             # Load CLIP
             dualcliploader = NODE_CLASS_MAPPINGS["DualCLIPLoader"]()
             dualcliploader_loaded = dualcliploader.load_clip(
                 clip_name1="t5xxl_fp16.safetensors",
                 clip_name2="ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors",
                 type="flux",
-                device="default"
+                device=device
             )
 
             # Text Encoding
@@ -202,7 +205,7 @@ def generate_image(prompt, input_image, lora_weight, guidance, downsampling_fact
         print(f"Error during generation: {str(e)}")
         return None
 
-# 7. Gradio Interface
+# 8. Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("# FLUX Redux Image Generator")
     
@@ -299,6 +302,5 @@ with gr.Blocks() as app:
     )
 
 if __name__ == "__main__":
-    # Download models if they don't exist
-    download_models()
+    # Download_models()
     app.launch(share=True)