Update app.py

app.py

@@ -59,6 +59,7 @@ ENABLE_CPU_OFFLOAD = 0
 BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+torch.set_default_device('cuda')
 
 style_list = [
     {