skytnt commited on
Commit
0f3c87f
1 Parent(s): 8851346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -101,8 +101,8 @@ class Model:
101
  w_avg = np.frombuffer(w_avg.raw_data, dtype=np.float32)[np.newaxis, :]
102
  w_avg = w_avg.repeat(16, axis=0)[np.newaxis, :]
103
  self.w_avg = w_avg
104
- self.g_mapping = rt.InferenceSession(g_mapping_path, providers=gpu_providers + providers)
105
- self.g_synthesis = rt.InferenceSession(g_synthesis_path, providers=gpu_providers + providers)
106
  self.encoder = rt.InferenceSession(encoder_path, providers=providers)
107
  self.detector = rt.InferenceSession(detector_path, providers=providers)
108
  detector_meta = self.detector.get_modelmeta().custom_metadata_map
 
101
  w_avg = np.frombuffer(w_avg.raw_data, dtype=np.float32)[np.newaxis, :]
102
  w_avg = w_avg.repeat(16, axis=0)[np.newaxis, :]
103
  self.w_avg = w_avg
104
+ self.g_mapping = rt.InferenceSession(g_mapping_path, providers=providers)
105
+ self.g_synthesis = rt.InferenceSession(g_synthesis_path, providers=providers)
106
  self.encoder = rt.InferenceSession(encoder_path, providers=providers)
107
  self.detector = rt.InferenceSession(detector_path, providers=providers)
108
  detector_meta = self.detector.get_modelmeta().custom_metadata_map