Update app.py
Browse files
app.py
CHANGED
|
@@ -14,11 +14,8 @@ pil_to_tensor = transforms.ToTensor()
|
|
| 14 |
# define some parameters based on the run we want to make
|
| 15 |
|
| 16 |
model = create_model()
|
| 17 |
-
|
| 18 |
checkpoints = torch.load('./weights/flolv2_UHDLL.pt', map_location=device)
|
| 19 |
-
|
| 20 |
model.load_state_dict(checkpoints['params'])
|
| 21 |
-
|
| 22 |
model = model.to(device)
|
| 23 |
|
| 24 |
def load_img (filename):
|
|
|
|
| 14 |
# define some parameters based on the run we want to make
|
| 15 |
|
| 16 |
model = create_model()
|
|
|
|
| 17 |
checkpoints = torch.load('./weights/flolv2_UHDLL.pt', map_location=device)
|
|
|
|
| 18 |
model.load_state_dict(checkpoints['params'])
|
|
|
|
| 19 |
model = model.to(device)
|
| 20 |
|
| 21 |
def load_img (filename):
|