update.
Browse filesEn 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")]
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 |
-
|
|
|
|
| 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 |
-
|
| 25 |
-
|
|
|
|
| 26 |
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|