Update app.py

app.py

@@ -3,15 +3,21 @@ from diffusers import DiffusionPipeline
 import torch
 import gradio as gr
 import os
+import torch
 
 # Faça login com seu token HF
 login(os.environ["HF_TOKEN"])  # use um token com permissão READ
 
+#Verificando se é CPU ou GPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if device == "cuda" else torch.float32
+
+
 # Carrega o modelo
 pipe = DiffusionPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell",
     use_auth_token=True,  # garante que use autenticação
-    torch_dtype=torch.float16
+    torch_dtype=dtype
 )
 
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")