nicolasrl commited on
Commit
2b19dce
verified
1 Parent(s): b110258

En la l铆nea de importaci贸n, se cambi贸 list_repo_files_info por list_repo_files.
En la llamada a la funci贸n, se usa list_repo_files(repo_id=repo_id).
Ajuste en el filtrado: La nueva funci贸n list_repo_files devuelve directamente una lista de rutas (strings), por lo que simplifiqu茅 el filtrado a [path for path in all_repo_files if path.endswith("model.onnx")]

Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -4,7 +4,8 @@ import onnxruntime as ort
4
  import numpy as np
5
  from PIL import Image
6
  import gradio as gr
7
- from huggingface_hub import hf_hub_download, list_repo_files_info, HfFolder
 
8
 
9
  # ----------------------------------------------------------------------
10
  # 1. Leer y guardar token de entorno
@@ -21,16 +22,18 @@ HfFolder.save_token(HF_TOKEN)
21
  repo_id = "nicolasrl/df-vs-real"
22
 
23
  print("Buscando archivos 'model.onnx' en el repositorio...")
24
- repo_files = list_repo_files_info(repo_id=repo_id)
25
- onnx_repo_paths = [f.path for f in repo_files if f.path.endswith("model.onnx")]
 
26
 
27
- if not onnx_repo_paths:
 
 
 
 
28
  raise FileNotFoundError("No se encontr贸 ning煤n archivo model.onnx en el repositorio.")
29
 
30
  latest_model_repo_path = max(onnx_repo_paths)
31
-
32
- # --- MODIFICACI脫N AQU脥 ---
33
- # Extraemos el nombre de la carpeta (versi贸n del modelo)
34
  model_version = os.path.dirname(latest_model_repo_path)
35
  print(f"Modelo m谩s reciente encontrado en el repo: {latest_model_repo_path} (Versi贸n: {model_version})")
36
 
@@ -85,8 +88,6 @@ demo = gr.Interface(
85
  gr.Textbox(label="Resultado")
86
  ],
87
  title="Detecci贸n de Deepfakes - ONNX",
88
- # --- MODIFICACI脫N AQU脥 ---
89
- # Usamos una f-string para incluir la versi贸n del modelo en la descripci贸n
90
  description=f"Modelo ONNX cargado desde la versi贸n m谩s reciente del repo.\n**Modelo en uso:** {model_version}"
91
  )
92
 
 
4
  import numpy as np
5
  from PIL import Image
6
  import gradio as gr
7
+ # --- CORRECCI脫N AQU脥 ---
8
+ from huggingface_hub import hf_hub_download, list_repo_files, HfFolder
9
 
10
  # ----------------------------------------------------------------------
11
  # 1. Leer y guardar token de entorno
 
22
  repo_id = "nicolasrl/df-vs-real"
23
 
24
  print("Buscando archivos 'model.onnx' en el repositorio...")
25
+ # --- CORRECCI脫N AQU脥 ---
26
+ # Usamos list_repo_files, que devuelve una lista de strings (rutas)
27
+ all_repo_files = list_repo_files(repo_id=repo_id)
28
 
29
+ # Filtramos la lista de rutas directamente
30
+ onnx_repo_paths = [path for path in all_repo_files if path.endswith("model.onnx")]
31
+
32
+
33
+ if not onx_repo_paths:
34
  raise FileNotFoundError("No se encontr贸 ning煤n archivo model.onnx en el repositorio.")
35
 
36
  latest_model_repo_path = max(onnx_repo_paths)
 
 
 
37
  model_version = os.path.dirname(latest_model_repo_path)
38
  print(f"Modelo m谩s reciente encontrado en el repo: {latest_model_repo_path} (Versi贸n: {model_version})")
39
 
 
88
  gr.Textbox(label="Resultado")
89
  ],
90
  title="Detecci贸n de Deepfakes - ONNX",
 
 
91
  description=f"Modelo ONNX cargado desde la versi贸n m谩s reciente del repo.\n**Modelo en uso:** {model_version}"
92
  )
93