Update app.py

app.py

@@ -176,7 +176,7 @@ transform = transforms.Compose([
     transforms.ToTensor(),  
     transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])  
 ])
-weight_dtype = torch.float32
+weight_dtype = torch.float16
 
 # line model
 line_model_path = os.path.join(model_global_path, 'LE', 'erika.pth')
@@ -201,7 +201,7 @@ global MultiResNetModel
 global cur_style
 
 cur_style = 'line + shadow'
-weight_dtype = torch.float32
+weight_dtype = torch.float16
 
 block_out_channels = [128, 128, 256, 512, 512]
 MultiResNetModel = MultiHiddenResNetModel(block_out_channels, len(block_out_channels))
@@ -313,7 +313,7 @@ print('loaded pipeline')
 
 @spaces.GPU
 def change_ckpt(style):
-    weight_dtype = torch.float32
+    weight_dtype = torch.float16
 
     if style == 'line':
         MultiResNetModel_path = os.path.join(model_global_path, 'line_GSRP', 'MultiResNetModel.bin')