Shailajagorile/MultiViewImage_Generator

1def inference_zero123(input_img: Image.Image, num_views: int = 4, save_prefix: str = "result"):2    """3    Generate multiple angle views for a given input image.4    Falls back to lower-res on CPU to avoid Hugging Face free tier timeouts.5    6    Args:7        input_img: PIL Image8        num_views: Number of different views to generate9        save_prefix: Prefix for output image files10    11    Returns:12        List of file paths for generated images13    """14    results = []15    for i in range(num_views):16        # Adjust resolution for CPU/GPU17        size = 512 if device == "cuda" else 25618 19        out = pipe(20            image=input_img,21            num_inference_steps=30 if device == "cuda" else 15,  # shorter steps on CPU22            guidance_scale=3.0,23            width=size,24            height=size,25        ).images[0]26 27        # Save result28        save_path = os.path.join(OUTPUT_DIR, f"{save_prefix}_view{i+1}.png")29        out.save(save_path)30        results.append(save_path)31 32    return results33