login client

app.py

@@ -2,7 +2,7 @@ import os
 import torch
 from diffusers import FluxPipeline # type: ignore
 import gradio as gr # type: ignore
-from huggingface_hub import login
+from huggingface_hub import login, InferenceClient
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
@@ -10,6 +10,11 @@ token = os.getenv("HF_TOKEN")
 login(token=token)
 
 
+client = InferenceClient(
+	provider="together",
+	api_key="token"
+)
+
 prompt = "A cat holding a sign that says hello world"
 image = pipe(
     prompt,
@@ -19,8 +24,9 @@ image = pipe(
     num_inference_steps=50,
     max_sequence_length=512,
     generator=torch.Generator("cpu").manual_seed(0),
-    use_auth_token=os.getenv("HF_TOKEN")
+    use_auth_token=token
 ).images[0]
+
 image.save("flux-dev.png")
 
 gradio_app = gr.Interface(