kd5678 commited on
Commit
200468f
·
verified ·
1 Parent(s): 08cb82c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -26,7 +26,8 @@ def pref_inpainting(image,
26
  with open("./configs/paintreward_train_configs.yaml") as file:
27
  config_dict= yaml.safe_load(file)
28
  config = munchify(config_dict)
29
-
 
30
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
31
  print(f"Current Device is {device}")
32
 
 
26
  with open("./configs/paintreward_train_configs.yaml") as file:
27
  config_dict= yaml.safe_load(file)
28
  config = munchify(config_dict)
29
+
30
+ print('Running here .........')
31
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
32
  print(f"Current Device is {device}")
33