Update app.py

app.py

@@ -71,7 +71,7 @@ def infer(audio_file):
     
     print(result)
 
-    images = pipe(prompt=result).images
+    images = pipe(prompt=result).images[0]
     
     #return cap_result, result, images
     return images