Update app.py

app.py

@@ -112,11 +112,11 @@ class Model:
         self.anime_seg = rt.InferenceSession(anime_seg_path, providers=providers)
 
     def get_img(self, w, noise=0):
-        img = self.g_synthesis.run(None, {'w': w, "noise": np.asarray([noise], dtype=np.float32)})[0]
+        img = self.g_synthesis.run(None, {'w': w.astype(np.float32), "noise": np.asarray([noise], dtype=np.float32)})[0]
         return (img.transpose(0, 2, 3, 1) * 127.5 + 128).clip(0, 255).astype(np.uint8)[0]
 
     def get_w(self, z, psi1, psi2):
-        return self.g_mapping.run(None, {'z': z, 'psi': np.asarray([psi1, psi2], dtype=np.float32)})[0]
+        return self.g_mapping.run(None, {'z': z.astype(np.float32), 'psi': np.asarray([psi1, psi2], dtype=np.float32)})[0]
 
     def remove_bg(self, img, s=1024):
         img0 = img