SpyC0der77 commited on
Commit
db506ab
·
verified ·
1 Parent(s): 36750d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
2
  import torch
3
  from diffusers import FluxPipeline
 
4
  from huggingface_hub import InferenceClient
5
  import os
6
 
 
7
  # Initialize the Flux pipeline
8
  def initialize_flux_pipeline():
9
  pipe = FluxPipeline.from_pretrained(
 
1
  import gradio as gr
2
  import torch
3
  from diffusers import FluxPipeline
4
+ import huggingface_hub
5
  from huggingface_hub import InferenceClient
6
  import os
7
 
8
+ huggingface_hub.login(token=os.getenv("HUGGINGFACE_API_TOKEN"))
9
  # Initialize the Flux pipeline
10
  def initialize_flux_pipeline():
11
  pipe = FluxPipeline.from_pretrained(