Emmylahot12 commited on
Commit
d6d554d
·
verified ·
1 Parent(s): c4c6e54

Create save_model.py

Browse files
Files changed (1) hide show
  1. save_model.py +17 -0
save_model.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import tensorflow as tf
3
+ from tensorflow.keras import layers, models
4
+
5
+ os.makedirs("model", exist_ok=True)
6
+
7
+ model = models.Sequential([
8
+ layers.Input(shape=(100,)),
9
+ layers.Dense(256, activation='relu'),
10
+ layers.Dense(128, activation='relu'),
11
+ layers.Dense(80)
12
+ ])
13
+
14
+ model.compile(optimizer='adam', loss='mse')
15
+ model.save("model/clone_tts_model.h5")
16
+
17
+ print("Model saved to model/clone_tts_model.h5")