Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,10 +2,7 @@ import torch
|
|
2 |
import gradio as gr
|
3 |
from diffusers import StableDiffusion3Pipeline
|
4 |
import os
|
5 |
-
from huggingface_hub import login
|
6 |
-
hf_token = os.getenv("HF_API_TOKEN")
|
7 |
|
8 |
-
login(token=hf_token)
|
9 |
def image_generator(prompt):
|
10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
11 |
pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers",
|
|
|
2 |
import gradio as gr
|
3 |
from diffusers import StableDiffusion3Pipeline
|
4 |
import os
|
|
|
|
|
5 |
|
|
|
6 |
def image_generator(prompt):
|
7 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
8 |
pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers",
|