Update feifeilib/feifeifluxapi.py

feifeilib/feifeifluxapi.py

@@ -3,6 +3,11 @@ from huggingface_hub import InferenceClient
 client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
 client.headers["x-use-cache"] = "0"
 
-def feifeifluxapi(protmpt):
-    image = client.text_to_image(protmpt)
+def feifeifluxapi(protmpt="the photo is a jpop girl",height=1152,width=896,guidance_scale=3.5):
+    image = client.text_to_image(
+        protmpt=protmpt,
+        height=height,
+        width=width,
+        guidance_scale=guidance_scale
+    )
     return image