Update app2.py

app2.py

@@ -15,22 +15,6 @@ variant = 'diffusion_transformer_768p'  # For high resolution
 current_directory = os.getcwd()
 model_path = os.path.join(current_directory, "pyramid_flow_model")  # Directory to store the model
 
-# Download the model if not already present
-def download_model_from_hf(model_repo, model_dir):
-    if not os.path.exists(model_dir):
-        print(f"Downloading model from {model_repo} to {model_dir}...")
-        snapshot_download(
-            repo_id=model_repo,
-            local_dir=model_dir,
-            local_dir_use_symlinks=False,
-            repo_type='model'
-        )
-        print("Model download complete.")
-    else:
-        print(f"Model directory '{model_dir}' already exists. Skipping download.")
-
-# Download model from Hugging Face if not present
-download_model_from_hf(model_repo, model_path)
 
 # Initialize model and move to CUDA
 torch.cuda.set_device(0)