Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,17 +6,14 @@ import os
|
|
6 |
|
7 |
# Fonction pour chercher le modèle dans tout le système de fichiers
|
8 |
def find_model_file(filename="generator.h5"):
|
9 |
-
# Parcourir tous les répertoires du serveur pour trouver le fichier
|
10 |
for root, dirs, files in os.walk("/"):
|
11 |
if filename in files:
|
12 |
-
|
13 |
-
return model_path
|
14 |
return None
|
15 |
|
16 |
# Fonction pour charger le modèle en toute sécurité
|
17 |
def load_model_safe(model_path):
|
18 |
try:
|
19 |
-
# Charger le modèle sans compilation pour éviter des erreurs liées au batch_shape
|
20 |
model = tf.keras.models.load_model(model_path, compile=False)
|
21 |
return model
|
22 |
except ValueError:
|
@@ -34,13 +31,10 @@ else:
|
|
34 |
# Fonction pour générer une vidéo à partir du générateur
|
35 |
def generate_video():
|
36 |
if generator is None:
|
37 |
-
#
|
38 |
error_file_path = "/content/error_file.txt"
|
39 |
-
|
40 |
with open(error_file_path, "w") as f:
|
41 |
-
|
42 |
-
|
43 |
-
# Retourner le fichier d'erreur
|
44 |
return error_file_path
|
45 |
|
46 |
# Générer un bruit aléatoire (entrée pour le générateur)
|
|
|
6 |
|
7 |
# Fonction pour chercher le modèle dans tout le système de fichiers
|
8 |
def find_model_file(filename="generator.h5"):
|
|
|
9 |
for root, dirs, files in os.walk("/"):
|
10 |
if filename in files:
|
11 |
+
return os.path.join(root, filename)
|
|
|
12 |
return None
|
13 |
|
14 |
# Fonction pour charger le modèle en toute sécurité
|
15 |
def load_model_safe(model_path):
|
16 |
try:
|
|
|
17 |
model = tf.keras.models.load_model(model_path, compile=False)
|
18 |
return model
|
19 |
except ValueError:
|
|
|
31 |
# Fonction pour générer une vidéo à partir du générateur
|
32 |
def generate_video():
|
33 |
if generator is None:
|
34 |
+
# Créer un fichier d'erreur générique sans message personnalisé
|
35 |
error_file_path = "/content/error_file.txt"
|
|
|
36 |
with open(error_file_path, "w") as f:
|
37 |
+
pass # Crée un fichier vide
|
|
|
|
|
38 |
return error_file_path
|
39 |
|
40 |
# Générer un bruit aléatoire (entrée pour le générateur)
|