Update Infer.py

Infer.py

@@ -1,21 +1,18 @@
 import tensorflow as tf
+import librosa
 import numpy as np
 import soundfile as sf
-import os
 
-MODEL_PATH = "model/clone_tts_model.h5"
-TEXT_MAX_LEN = 100
-SAMPLE_RATE = 22050
+# Load the trained model
+model = tf.keras.models.load_model('model/clone_tts_model.h5')
 
-def generate_speech(text_input):
-    x_input = np.array([[ord(c) for c in text_input.ljust(TEXT_MAX_LEN)[:TEXT_MAX_LEN]]])
-    model = tf.keras.models.load_model(MODEL_PATH)
-    audio = model.predict(x_input)[0]
-    os.makedirs("output", exist_ok=True)
-    output_path = "output/generated.wav"
-    sf.write(output_path, audio, SAMPLE_RATE)
-    print(f"Generated speech saved at: {output_path}")
+# Define input text
+text_input = "Hello, welcome to CloneTTS. This is an example of text-to-speech synthesis."
 
-if __name__ == "__main__":
-    text = input("Enter text to synthesize: ")
-    generate_speech(text)
+# Generate the speech (preprocess as needed depending on model requirements)
+speech = model.predict(np.array([text_input]))
+
+# Save the generated speech to a .wav file
+sf.write('output/speech.wav', speech, 22050)  # Adjust sample rate as necessary
+
+print("Speech generated and saved as 'output/speech.wav'")