Update app.py

app.py

@@ -28,7 +28,7 @@ def predict(text):
     with torch.autocast("cuda"):
         outputs = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=512)
 
-    with tempfile.NamedTemporaryFile() as tmp:
+    with tempfile.NamedTemporaryFile(suffix='.mp4') as tmp:
         video_path = gr.make_waveform((32000, outputs[0].cpu().numpy().astype(np.float16).ravel()), bars_color=random.choice(COLORS), bar_count=75)
         command = f'ffmpeg -y -i {video_path} -vf "scale=250:150" {tmp.name}'
         subprocess.run(command.split())