Update app.py

app.py

@@ -41,7 +41,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 from diffusers import DiffusionPipeline
 
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype="auto")
 lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
 trigger_word = "Super Realism"  # Leave blank if no trigger word is needed.
 pipe.load_lora_weights(lora_repo)
@@ -73,7 +73,7 @@ STYLE_NAMES = list(styles.keys())
 def apply_style(style_name: str, positive: str) -> str:
     return styles.get(style_name, styles[DEFAULT_STYLE_NAME]).replace("{prompt}", positive)
 
-@spaces.GPU(duration=60, enable_queue=True)
+@spaces.GPU
 def generate_image_flux(
     prompt: str,
     seed: int = 0,
@@ -118,7 +118,7 @@ smol_processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM2-2.2B-Inst
 smol_model = AutoModelForImageTextToText.from_pretrained(
     "HuggingFaceTB/SmolVLM2-2.2B-Instruct", 
     _attn_implementation="flash_attention_2",
-    torch_dtype=torch.bfloat16
+    torch_dtype="auto"
 ).to("cuda:0")
 
 # -------------------------------