Socold commited on
Commit
e8e9070
·
verified ·
1 Parent(s): d7aaa33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -46
app.py CHANGED
@@ -18,55 +18,41 @@ subprocess.run([sys.executable, "-m", "pip", "install", "-e", "VV_Clone_1"])
18
  import torch
19
  import gradio as gr
20
 
21
- # FORCER la récupération des secrets
 
 
 
22
  USERNAME = os.environ.get("AUTH_USERNAME")
23
  PASSWORD = os.environ.get("AUTH_PASSWORD")
24
 
25
- if not USERNAME or not PASSWORD:
26
- # Si pas de secrets, bloquer complètement
27
- gr.Interface(
28
- fn=lambda x: "❌ ERREUR: Configurez AUTH_USERNAME et AUTH_PASSWORD dans Settings → Variables and secrets!",
29
- inputs="text",
30
- outputs="text",
31
- title="Space non configuré"
32
- ).launch()
33
- exit()
34
-
35
- print(f"✅ Authentification activée pour: {USERNAME}")
36
-
37
- # Créer un nouveau script qui wrappe tout avec auth
38
- wrapper_script = f"""
39
- import sys
40
- sys.path.insert(0, "VV_Clone_1")
41
- os.chdir("VV_Clone_1")
42
-
43
- from demo.gradio_demo import VibeVoiceDemo
44
- import gradio as gr
45
-
46
- # Créer l'instance
47
- demo = VibeVoiceDemo(model_path="{MODEL_PATH}")
48
-
49
- # Créer l'interface Gradio
50
- interface = demo.launch()
51
-
52
- # Fermer l'instance sans auth
53
- interface.close()
54
-
55
- # Relancer AVEC authentification
56
- interface.launch(
57
- auth=("{USERNAME}", "{PASSWORD}"),
58
- auth_message="Entrez vos identifiants VibeVoice",
59
- server_name="0.0.0.0",
60
- server_port=7860,
61
- share=False
62
- )
63
- """
64
-
65
- MODEL_PATH = "aoi-ot/VibeVoice-7B"
66
 
67
- # Écrire et exécuter le wrapper
68
- with open("run_with_auth.py", "w") as f:
69
- f.write(wrapper_script)
 
 
70
 
 
71
  os.chdir("VV_Clone_1")
72
- exec(open("../run_with_auth.py").read())
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
  import torch
19
  import gradio as gr
20
 
21
+ # Configuration
22
+ MODEL_PATH = "aoi-ot/VibeVoice-7B"
23
+
24
+ # Récupérer les secrets
25
  USERNAME = os.environ.get("AUTH_USERNAME")
26
  PASSWORD = os.environ.get("AUTH_PASSWORD")
27
 
28
+ print(f"Chargement du modèle: {MODEL_PATH}")
29
+ print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
 
31
+ # Vérifier l'auth
32
+ if USERNAME and PASSWORD:
33
+ print(f"✅ Authentification activée pour: {USERNAME}")
34
+ else:
35
+ print("⚠️ Pas d'authentification configurée")
36
 
37
+ # Lancer le script principal
38
  os.chdir("VV_Clone_1")
39
+ sys.argv = ["gradio_demo.py", "--model_path", MODEL_PATH, "--share"]
40
+
41
+ # Modifier le fichier pour ajouter l'auth
42
+ if USERNAME and PASSWORD:
43
+ with open("demo/gradio_demo.py", "r") as f:
44
+ content = f.read()
45
+
46
+ # Remplacer la ligne launch
47
+ content = content.replace(
48
+ 'demo.launch(share=args.share)',
49
+ f'demo.launch(share=args.share, auth=("{USERNAME}", "{PASSWORD}"))'
50
+ )
51
+
52
+ with open("demo/gradio_demo.py", "w") as f:
53
+ f.write(content)
54
+
55
+ # Importer et lancer
56
+ from demo.gradio_demo import main
57
+ print("Démarrage de VibeVoice-7B avec authentification...")
58
+ main()