Delete inference.py

inference.py

@@ -1,90 +0,0 @@
-#!/usr/bin/env python3
-"""
-Simple inference script for background removal using HuggingFace Inference API
-"""
-
-import os
-import sys
-from pathlib import Path
-from huggingface_hub import InferenceClient
-from PIL import Image
-
-def remove_background(input_path, output_path, model_id=None, hf_token=None):
-    """
-    Remove background from an image using HuggingFace Inference API
-    
-    Args:
-        input_path (str): Path to input image
-        output_path (str): Path to save output image
-        model_id (str): HuggingFace model ID
-        hf_token (str): HuggingFace API token
-    """
-    
-    # Get token and model from environment if not provided
-    if not hf_token:
-        hf_token = os.getenv("HF_TOKEN")
-    if not hf_token:
-        raise ValueError("HF_TOKEN is required. Set it as environment variable or pass as argument.")
-    
-    if not model_id:
-        model_id = os.getenv("MODEL_ID", "briaai/RMBG-1.4")
-    
-    # Initialize client
-    client = InferenceClient(token=hf_token)
-    
-    # Load image
-    if not os.path.exists(input_path):
-        raise FileNotFoundError(f"Input image not found: {input_path}")
-    
-    print(f"Loading image: {input_path}")
-    image = Image.open(input_path)
-    
-    print(f"Processing with model: {model_id}")
-    try:
-        # Process image
-        result = client.image_segmentation(image, model=model_id)
-        
-        # Handle different output formats
-        if isinstance(result, list) and len(result) > 0:
-            # If model returns segmentation masks
-            output_image = result[0]['mask']
-        else:
-            # If model returns direct image
-            output_image = result
-        
-        # Save result
-        if hasattr(output_image, 'save'):
-            output_image.save(output_path)
-            print(f"Background removed image saved: {output_path}")
-        else:
-            print("Unexpected output format from model")
-            return False
-            
-        return True
-        
-    except Exception as e:
-        print(f"Error processing image: {str(e)}")
-        return False
-
-def main():
-    """Command line interface"""
-    if len(sys.argv) < 3:
-        print("Usage: python inference.py <input_image> <output_image> [model_id]")
-        print("Example: python inference.py input.jpg output.png")
-        print("Example: python inference.py input.jpg output.png briaai/RMBG-1.4")
-        sys.exit(1)
-    
-    input_path = sys.argv[1]
-    output_path = sys.argv[2]
-    model_id = sys.argv[3] if len(sys.argv) > 3 else None
-    
-    success = remove_background(input_path, output_path, model_id)
-    
-    if success:
-        print("✅ Background removal completed successfully!")
-    else:
-        print("❌ Background removal failed!")
-        sys.exit(1)
-
-if __name__ == "__main__":
-    main()