Dannyar608 commited on
Commit
42d6a19
·
verified ·
1 Parent(s): 60c1c53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -1,6 +1,14 @@
1
  import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
 
 
 
 
 
 
 
 
4
 
5
  # Load the model (you can replace this with the specific Kwai-Kolors/Kolors model if available)
6
  pipe = StableDiffusionPipeline.from_pretrained("Kwai-Kolors/Kolors")
 
1
  import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
+ import os
5
+
6
+ token = os.getenv('token')
7
+
8
+ from huggingface_hub import login
9
+ login(token=token)
10
+
11
+ gr.load("models/black-forest-labs/FLUX.1-dev").launch()
12
 
13
  # Load the model (you can replace this with the specific Kwai-Kolors/Kolors model if available)
14
  pipe = StableDiffusionPipeline.from_pretrained("Kwai-Kolors/Kolors")