Delete flux_api.py

flux_api.py

@@ -1,59 +0,0 @@
-
-import argparse
-import requests
-from gradio_client import Client, handle_file
-
-def generate_image(prompt, image_url=None, image_file=None):
-    client = Client("yanze/PuLID-FLUX")
-
-    # Determine input image
-    if image_url:
-        id_image = handle_file(image_url)
-    elif image_file:
-        id_image = handle_file(image_file)
-    else:
-        return "Error: Please provide an image URL or upload an image file."
-
-    try:
-        result = client.predict(
-            prompt=prompt,
-            id_image=id_image,
-            start_step=0,
-            guidance=4,
-            seed="-1",
-            true_cfg=1,
-            width=896,
-            height=1152,
-            num_steps=20,
-            id_weight=1,
-            neg_prompt="bad quality, worst quality, text, signature, watermark, extra limbs",
-            timestep_to_start_cfg=1,
-            max_sequence_length=128,
-            api_name="/generate_image"
-        )
-
-        base_url = "https://yanze-pulid-flux.hf.space/file="
-        file_path = result[0]
-        full_url = f"{base_url}{file_path}"
-
-        # Download and save image
-        response = requests.get(full_url)
-        if response.status_code == 200:
-            with open("output_image.jpg", "wb") as f:
-                f.write(response.content)
-            print("Image saved as output_image.jpg")
-        
-        return full_url
-    except Exception as e:
-        return f"Error during prediction: {str(e)}"
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Generate images using PuLID-FLUX")
-    parser.add_argument("--prompt", type=str, required=True, help="Text prompt for the image generation")
-    parser.add_argument("--image_url", type=str, default=None, help="Optional image URL")
-    parser.add_argument("--image_file", type=str, default=None, help="Optional local image file path")
-    
-    args = parser.parse_args()
-    
-    output_url = generate_image(args.prompt, args.image_url, args.image_file)
-    print(f"Generated Image URL: {output_url}")