Delete huggingface_hub.py

huggingface_hub.py

@@ -1,28 +0,0 @@
-from huggingface_hub import hf_hub_download
-from safetensors.torch import load_file
-from diffusers import StableDiffusionPipeline
-
-def load_pipeline(base_model_path, lora_repo_id, lora_filename):
-    # Load the base SD pipeline
-    pipeline = StableDiffusionPipeline.from_pretrained(base_model_path)
-
-    # Download the LoRA file
-    lora_path = hf_hub_download(repo_id=lora_repo_id, filename=lora_filename)
-    
-    # Load the LoRA weights
-    lora_state_dict = load_file(lora_path)
-    pipeline.unet.load_attn_procs(lora_state_dict)
-
-    return pipeline
-
-# Define parameters
-base_model_path = "runwayml/stable-diffusion-v1-5"
-lora_repo_id = "maria26/Floor_Plan_LoRA"
-lora_filename = "model.safetensors"
-
-# Load the pipeline
-pipeline = load_pipeline(base_model_path, lora_repo_id, lora_filename)
-
-def predict(prompt):
-    result = pipeline(prompt).images[0]
-    return result