fix dtype in migration

app.py

@@ -194,7 +194,8 @@ You should use {formatted_words} to trigger the image generation.
     url: >-
       {image}
 """
-
+    dtype = "torch.bfloat16" if info["baseModel"] == "black-forest-labs/FLUX.1-dev" or info["baseModel"] == "black-forest-labs/FLUX.1-schnell" else "torch.float16"
+        
     content = f"""---
 license: other
 license_name: bespoke-lora-trained-license
@@ -234,7 +235,9 @@ Weights for this model are available in Safetensors format.
 from diffusers import AutoPipelineForText2Image
 import torch
 
-pipeline = AutoPipelineForText2Image.from_pretrained('{info["baseModel"]}', torch_dtype=torch.float16).to('cuda')
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+pipeline = AutoPipelineForText2Image.from_pretrained('{info["baseModel"]}', torch_dtype={dtype}).to(device)
 pipeline.load_lora_weights('{user_repo_id}', weight_name='{downloaded_files["weightName"][0]}')
 image = pipeline('{prompt if prompt else (formatted_words if formatted_words else 'Your custom prompt')}').images[0]
 ```