Update feifeilib/feifeifluxapi.py

feifeilib/feifeifluxapi.py

@@ -1,27 +1,18 @@
 import os
-# from huggingface_hub import InferenceClient
-from gradio_client import Client
+from huggingface_hub import InferenceClient
 
-
-#client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
-#client.headers["x-use-cache"] = "0"
+client = InferenceClient("aifeifei798/feifei-flux-lora-v1.1", token=os.getenv('HF_TOKEN'))
+client.headers["x-use-cache"] = "0"
 
 def feifeifluxapi(prompt, height=1152, width=896, guidance_scale=3.5):
-#    image = client.text_to_image(
-#        prompt=prompt,
-#        height=height,
-#        width=width,
-#        guidance_scale=guidance_scale
-#    )
-#    return image
-    client = Client("aifeifei798/DarkIdol-flux")
-    result = client.predict(
+    # output is a PIL.Image object
+    result = client.text_to_image(
     		prompt=prompt,
     		seed=0,
     		randomize_seed=True,
     		width=width,
     		height=height,
     		num_inference_steps=6,
-    		api_name="/infer"
+    		guidance_scale=guidance_scale
     )
     return result