Update app.py

app.py

@@ -137,7 +137,7 @@ def generate_end_frame(start_frame, edit_prompt: str):
             api_name="/infer"
         )
 
-        return result['image']
+        return result[0]['image']
 
     except Exception as e:
         import traceback