vcollos commited on
Commit
8c40186
·
1 Parent(s): 662cc18

Add application file

Browse files
Files changed (1) hide show
  1. app.py +9 -1
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(base_model, torch_dtype=torch.bfloat16)
 
 
 
 
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 = ""