Update app.py

app.py

@@ -19,17 +19,23 @@ 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)
 
-# Generate Comedy Script
+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):
-    response = client.chat.completions.create(
+    script = client.chat.completions.create(
         messages=[
-            {"role": "user", "content": prompt}
+            {"role": "user", "content": write a comedy scipt on {prompt}}
         ],
-        model="llama-3.1"
-    )
-    script = response["choices"][0]["message"]["content"]
+        model="llama3-8b-8192"
+    )[0]['generated_text']
     return script
-
 # Convert Text to Speech using Coqui TTS
 def text_to_speech(script):
     output_audio = 'output.wav'