inversed_model_dict switch to dict comprehension

#6
by newturok - opened
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -46,9 +46,7 @@ huggingface_hub.hf_hub_download(
46
  'pexels-cottonbro-studio-6649832-960x506-25fps.mp4',
47
  local_dir='videos')
48
 
49
- inversed_model_dict = dict()
50
- for k, v in model_dict.items():
51
- inversed_model_dict[v] = k
52
 
53
  to_tensor = T.PILToTensor()
54
  blur = T.GaussianBlur(kernel_size=(9, 9), sigma=(18, 18))
 
46
  'pexels-cottonbro-studio-6649832-960x506-25fps.mp4',
47
  local_dir='videos')
48
 
49
+ inversed_model_dict = {v: k for k, v in model_dict.items()}
 
 
50
 
51
  to_tensor = T.PILToTensor()
52
  blur = T.GaussianBlur(kernel_size=(9, 9), sigma=(18, 18))