Socold commited on
Commit
92e3233
·
verified ·
1 Parent(s): e05137d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -11
app.py CHANGED
@@ -19,22 +19,20 @@ import torch
19
  import gradio as gr
20
  from demo.gradio_demo import VibeVoiceDemo
21
 
22
- # Configuration
 
 
 
23
  MODEL_PATH = "aoi-ot/VibeVoice-Large"
24
- USERNAME = "socold01"
25
- PASSWORD = "V#ztdUeS4Zp$0Z4ZB^y3DWhJ"
26
 
27
  print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
28
- print(f"VRAM: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.2f} GB")
29
 
30
- # Créer l'instance et la lancer DIRECTEMENT
31
- demo_instance = VibeVoiceDemo(
32
- model_path=MODEL_PATH,
33
- device="cuda" if torch.cuda.is_available() else "cpu"
34
- )
35
 
36
- # Lancer directement avec l'authentification
37
- demo_instance.launch(
38
  auth=(USERNAME, PASSWORD),
39
  server_name="0.0.0.0",
40
  share=False
 
19
  import gradio as gr
20
  from demo.gradio_demo import VibeVoiceDemo
21
 
22
+ # Récupérer les secrets (invisibles publiquement)
23
+ USERNAME = os.environ["AUTH_USERNAME"] # Crash si pas défini
24
+ PASSWORD = os.environ["AUTH_PASSWORD"]
25
+
26
  MODEL_PATH = "aoi-ot/VibeVoice-Large"
 
 
27
 
28
  print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
 
29
 
30
+ # Créer et lancer
31
+ demo = VibeVoiceDemo(model_path=MODEL_PATH)
32
+ interface = demo.launch_demo()
 
 
33
 
34
+ # Lancer avec authentification depuis les secrets
35
+ interface.launch(
36
  auth=(USERNAME, PASSWORD),
37
  server_name="0.0.0.0",
38
  share=False