MohamedRashad commited on
Commit
e5b7e3e
·
1 Parent(s): 14626ab

Fix indentation in load_infinity function for better readability

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -220,10 +220,10 @@ def load_infinity(
220
  state_dict = torch.load(model_path, map_location=device)
221
  print(infinity_test.load_state_dict(state_dict))
222
 
223
- # Initialize random number generator on the correct device
224
- infinity_test.rng = torch.Generator(device=device)
225
 
226
- return infinity_test
227
 
228
  def transform(pil_img, tgt_h, tgt_w):
229
  width, height = pil_img.size
 
220
  state_dict = torch.load(model_path, map_location=device)
221
  print(infinity_test.load_state_dict(state_dict))
222
 
223
+ # Initialize random number generator on the correct device
224
+ infinity_test.rng = torch.Generator(device=device)
225
 
226
+ return infinity_test
227
 
228
  def transform(pil_img, tgt_h, tgt_w):
229
  width, height = pil_img.size