Update app.py

app.py

@@ -32,9 +32,9 @@ def process_adain_api(files, style_strength=1.0, dataset_size=100):
         # Create zip file
         zip_path = os.path.join(temp_dir, "style_transfer_results.zip")
         with zipfile.ZipFile(zip_path, 'w') as zipf:
-            for file in os.listdir(result_path):
+            for file in os.listdir(output_dir):
                 if file.lower().endswith(('.jpg', '.jpeg', '.png')):
-                    zipf.write(os.path.join(result_path, file), file)
+                    zipf.write(os.path.join(output_dir, file), file)
         
         return zip_path, f"Style transfer completed with strength {style_strength}"