Update app.py

app.py

@@ -19,22 +19,13 @@ image_generator = StableDiffusionPipeline.from_pretrained("stabilityai/stable-di
 # Use Coqui TTS (open-source) for text-to-speech
 tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False, gpu=False)
 
-chat_completion = client.chat.completions.create(
-    messages=[
-        {
-            "role": "user",
-            "content": "Explain the importance of fast language models",
-        }
-    ],
-    model="llama3-8b-8192",
-)
 def generate_comedy_script(prompt):
     script = client.chat.completions.create(
         messages=[
             {"role": "user", "content": "write a comedy scipt on {prompt}"}
         ],
         model="llama3-8b-8192"
-    )[0]['generated_text']
+    script = response['choices'][0]['message']['content']
     return script
 # Convert Text to Speech using Coqui TTS
 def text_to_speech(script):