Update app.py

app.py

@@ -43,7 +43,7 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda
 #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 torch.cuda.empty_cache()
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 model_id = ("zer0int/LongCLIP-GmP-ViT-L-14")
 config = CLIPConfig.from_pretrained(model_id)