Kiwinicki commited on
Commit
7c184f5
·
1 Parent(s): 70197b6

extend transforms and change requirements.txt encoding

Browse files
Files changed (2) hide show
  1. handler.py +7 -3
  2. requirements.txt +0 -0
handler.py CHANGED
@@ -13,9 +13,13 @@ from model import Generator
13
  class EndpointHandler:
14
 
15
  def __init__(self, path=''):
16
- self.transform = transforms.Compose([
17
- transforms.ToTensor()
18
- ])
 
 
 
 
19
 
20
  repo_id = "Kiwinicki/sat2map-generator"
21
  generator_path = hf_hub_download(repo_id=repo_id, filename="generator.pth")
 
13
  class EndpointHandler:
14
 
15
  def __init__(self, path=''):
16
+ self.transform = transforms.Compose(
17
+ [
18
+ transforms.ToImage(),
19
+ transforms.ToDtype(torch.float32, scale=True),
20
+ transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
21
+ ]
22
+ )
23
 
24
  repo_id = "Kiwinicki/sat2map-generator"
25
  generator_path = hf_hub_download(repo_id=repo_id, filename="generator.pth")
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ