Update app.py

app.py

@@ -11,10 +11,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 hf_token = os.getenv('HF_TOKEN')
 
 if torch.cuda.is_available():
-    torch_dtype = torch.float16
+    dtype = torch.float16
     torch.cuda.empty_cache()
 else:
-    torch_dtype = torch.float32
+    dtype = torch.float32
 
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
 pipe = pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", token=hf_token, torch_dtype=torch.bfloat16)