Update app.py

app.py

@@ -1,16 +1,19 @@
 import gradio as gr
-#import spaces
+import spaces
 from diffusers import DiffusionPipeline
 import torch
 
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+
 pipe = DiffusionPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
-    torch_dtype=torch.float16
-)
-pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors")
-pipe = pipe.to("cuda")
+    torch_dtype=torch.bfloat16
+).to(device)
+pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors").to(device)
 
-#@spaces.GPU(duration=75)
+@spaces.GPU(duration=75)
 def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):
     """Generates an image using the FLUX.1-dev LoRA model."""
     generator = torch.Generator("cuda").manual_seed(seed) if seed else None