Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 import tempfile
 import zipfile
-from your_adain_module import run_adain  # Your actual AdaIN implementation
+from test_api import run_adain
 
 def process_adain_api(files, style_strength=1.0):
     '''API endpoint for AdaIN processing.'''
@@ -23,11 +23,10 @@ def process_adain_api(files, style_strength=1.0):
             shutil.copy(file.name, input_dir)
     
     try:
-        # Your internal style images path
-        style_images_path = "path/to/your/style/images"
+        style_dataset_pth = "tidalove/paleo-real"
         
         # Run AdaIN
-        result_path = run_adain(input_dir, style_images_path, output_dir, style_strength)
+        result_path = run_adain(input_dir, style_dataset_pth, output_dir, style_strength)
         
         # Create zip file
         zip_path = os.path.join(temp_dir, "style_transfer_results.zip")