Update main.py

main.py

@@ -51,45 +51,7 @@ def generate_image_caption(image_path):
         out = sd_pipe(img_tensor, guidance_scale=3)
         out["images"][0].save("img1.jpg")
 
-    # Blip image captioning
-    raw_image = Image.open(image_path).convert("RGB")
     
-    processor = BlipProcessor.from_pretrained(
-        "Salesforce/blip-image-captioning-large"
-    )
-    model = BlipForConditionalGeneration.from_pretrained(
-        "Salesforce/blip-image-captioning-large"
-    ).to(device)
-    
-    # Conditional image captioning
-    text = "a photography of"
-    inputs = processor(raw_image, text, return_tensors="pt").to(device)
-    out = model.generate(**inputs)
-    caption = processor.decode(out[0], skip_special_tokens=True)
-    
-    # Unconditional image captioning
-    inputs = processor(raw_image, return_tensors="pt").to(device)
-    out = model.generate(**inputs)
-    caption = processor.decode(out[0], skip_special_tokens=True)
-
-    # Stable diffusion pipeline
-    model_id = "prompthero/openjourney"
-    pipe = StableDiffusionPipeline.from_pretrained(
-        model_id, torch_dtype=torch.float32
-    )
-    pipe = pipe.to(device)
-
-    Room = "Living Room"
-    AI_Intervention = "High"
-    Mode = "Redesign"
-    Design = "Modern"
-    prompt = (
-        f"Give me a realistic and complete image of {caption} "
-        f"which room type: {Room}, AI Intervention: {AI_Intervention}, "
-        f"Mode: {Mode} and Design style: {Design}"
-    )
-    image = pipe(prompt).images[0]
-    image.save("result3.jpg")