Spaces:
Running
Running
Update app.py
Browse files
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=
|
105 |
-
self.g_synthesis = rt.InferenceSession(g_synthesis_path, 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
|