Update app.py

app.py

@@ -7,34 +7,6 @@ from pulid import attention_processor as attention
 from pulid.pipeline import PuLIDPipeline
 from pulid.utils import resize_numpy_image_long, seed_everything
 
-print(torch.__version__)
-
-import shutil
-
-def find_cuda():
-    # Check if CUDA_HOME or CUDA_PATH environment variables are set
-    cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
-
-    if cuda_home and os.path.exists(cuda_home):
-        return cuda_home
-
-    # Search for the nvcc executable in the system's PATH
-    nvcc_path = shutil.which('nvcc')
-
-    if nvcc_path:
-        # Remove the 'bin/nvcc' part to get the CUDA installation path
-        cuda_path = os.path.dirname(os.path.dirname(nvcc_path))
-        return cuda_path
-
-    return None
-
-cuda_path = find_cuda()
-
-if cuda_path:
-    print(f"CUDA installation found at: {cuda_path}")
-else:
-    print("CUDA installation not found")
-
 
 pipeline = PuLIDPipeline()
 
@@ -151,7 +123,7 @@ with gr.Blocks(title="PuLID", css=".gr-box {border-color: #8136e2}") as demo:
                 maximum=1.5,
                 step=0.1,
             )
-            n_samples = gr.Slider(label="Num samples", value=4, minimum=1, maximum=8, step=1)
+            n_samples = gr.Slider(label="Num samples", value=4, minimum=1, maximum=4, step=1)
             seed = gr.Slider(
                 label="Seed", value=42, minimum=np.iinfo(np.uint32).min, maximum=np.iinfo(np.uint32).max, step=1
             )