Add application file
Browse files
app.py
CHANGED
@@ -8,15 +8,23 @@ from datetime import datetime
|
|
8 |
import json
|
9 |
from gradio_client import Client as client_gradio
|
10 |
from supabase import create_client, Client
|
|
|
11 |
|
12 |
# Inicializar Supabase com variáveis de ambiente
|
13 |
SUPABASE_URL = os.getenv("SUPABASE_URL")
|
14 |
SUPABASE_KEY = os.getenv("SUPABASE_KEY")
|
15 |
supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
|
16 |
|
|
|
|
|
|
|
17 |
# Carregar modelo base e LoRA
|
18 |
base_model = "black-forest-labs/FLUX.1-dev"
|
19 |
-
pipe = DiffusionPipeline.from_pretrained(
|
|
|
|
|
|
|
|
|
20 |
|
21 |
lora_repo = "markury/AndroFlux"
|
22 |
trigger_word = ""
|
|
|
8 |
import json
|
9 |
from gradio_client import Client as client_gradio
|
10 |
from supabase import create_client, Client
|
11 |
+
from huggingface_hub import login
|
12 |
|
13 |
# Inicializar Supabase com variáveis de ambiente
|
14 |
SUPABASE_URL = os.getenv("SUPABASE_URL")
|
15 |
SUPABASE_KEY = os.getenv("SUPABASE_KEY")
|
16 |
supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
|
17 |
|
18 |
+
# Autenticar com o Hugging Face
|
19 |
+
login(token="HF_TOKEN")
|
20 |
+
|
21 |
# Carregar modelo base e LoRA
|
22 |
base_model = "black-forest-labs/FLUX.1-dev"
|
23 |
+
pipe = DiffusionPipeline.from_pretrained(
|
24 |
+
base_model,
|
25 |
+
torch_dtype=torch.bfloat16,
|
26 |
+
use_auth_token=True
|
27 |
+
)
|
28 |
|
29 |
lora_repo = "markury/AndroFlux"
|
30 |
trigger_word = ""
|