Upload folder using huggingface_hub

app.py

@@ -8,7 +8,7 @@ import spaces
 
 repo = "stabilityai/stable-diffusion-3-medium-diffusers"
 if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
+    pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to("cuda")
 else:
     pipe = StableDiffusion3Pipeline.from_pretrained(repo)