Leo8613 commited on
Commit
6406a90
·
verified ·
1 Parent(s): b389d1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
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
- model_path = os.path.join(root, filename)
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
- # Retourner un fichier d'erreur générique si le modèle n'est pas trouvé
38
  error_file_path = "/content/error_file.txt"
39
-
40
  with open(error_file_path, "w") as f:
41
- f.write("Le modèle n'a pas pu être chargé.")
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)