Rename inference.py to handler.py

inference.py → handler.py

@@ -5,7 +5,7 @@ from io import BytesIO
 
 class ImageGenerator:
     def __init__(self):
-        self.model_base = "femboysLover/blue_pencil-fp16-XL"
+        self.model_base = "AIhgenerator/nsfwxxl2"
         self.v_autoencoder = "madebyollin/sdxl-vae-fp16-fix"
         self.model_refiner = "stabilityai/stable-diffusion-xl-refiner-1.0"
 
@@ -54,11 +54,3 @@ class ImageGenerator:
         image_refiner.save(buffer, format="JPEG")
         return buffer.getvalue()
 
-# Usage example
-image_generator = ImageGenerator()
-result = image_generator.generate_image(
-    prompt="A description of the image you want to generate",
-    prompt2="Additional description if needed",
-    negative_prompt="What you want to avoid in the image",
-    negative_prompt2="Additional negative prompt if needed"
-)