Update app.py
Browse files
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 |
|