Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -167,8 +167,6 @@ def generate_multi_model(input_img):
|
|
| 167 |
|
| 168 |
|
| 169 |
def generate_one_shot(src_img, img_prompt):
|
| 170 |
-
output_img = src_img
|
| 171 |
-
return output_img
|
| 172 |
|
| 173 |
# init model
|
| 174 |
state_dict = torch.load(f"./checkpoints/{img_prompt[-2:]}/epoch_latest.pth", map_location='cpu')
|
|
|
|
| 167 |
|
| 168 |
|
| 169 |
def generate_one_shot(src_img, img_prompt):
|
|
|
|
|
|
|
| 170 |
|
| 171 |
# init model
|
| 172 |
state_dict = torch.load(f"./checkpoints/{img_prompt[-2:]}/epoch_latest.pth", map_location='cpu')
|