Update main.py

main.py

@@ -1,7 +1,14 @@
-from fastapi import FastAPI
+from bark import SAMPLE_RATE, generate_audio, preload_models
+from scipy.io.wavfile import write as write_wav
+# download and load all models
+preload_models()
 
-app = FastAPI()
+# generate audio from text
+text_prompt = """
+     Hello, my name is Suno. And, uh — and I like pizza. [laughs] 
+     But I also have other interests such as playing tic tac toe.
+"""
+speech_array = generate_audio(text_prompt)
 
+write_wav("/path/to/audio.wav", SAMPLE_RATE, audio_array)
 
-def read_root():
-    return {"Hello": "World!"}