Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 # Function to generate and display image
 def generate_and_display_image(prompt):
     # Initialize the UNet model
-    unet = UNet2DConditionModel.from_pretrained("https://huggingface.co/spaces/lakshmikarpolam/LLCM_Medical_Images_Generation/tree/main/unet", torch_dtype=torch.float16, variant="fp16")
+    unet = UNet2DConditionModel.from_pretrained("./unet", torch_dtype=torch.float16, variant="fp16")
 
     # Initialize the diffusion pipeline
     pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", unet=unet, torch_dtype=torch.float16, variant="fp16")