Spaces:
Runtime error
Runtime error
modifico la lectura de clases
Browse files
app.py
CHANGED
@@ -18,9 +18,20 @@ transform_preprocess = transforms.Compose([
|
|
18 |
])
|
19 |
|
20 |
# Cargar el conjunto de datos Food-101 para obtener la lista de clases
|
21 |
-
image_path = '../../compartida/vision-project/'
|
22 |
-
food101_dataset = datasets.Food101(image_path, split='train')
|
23 |
-
classes = food101_dataset.classes
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
# Funci贸n para predecir la clase de una nueva imagen
|
26 |
def predict_image(image):
|
|
|
18 |
])
|
19 |
|
20 |
# Cargar el conjunto de datos Food-101 para obtener la lista de clases
|
21 |
+
#image_path = '../../compartida/vision-project/'
|
22 |
+
#food101_dataset = datasets.Food101(image_path, split='train')
|
23 |
+
#classes = food101_dataset.classes
|
24 |
+
|
25 |
+
from datasets import load_remote_dataset
|
26 |
+
# URL del archivo de texto en el Space de Hugging Face
|
27 |
+
file_url = "https://huggingface.co/datasets/username/dataset_name/raw/main/classes.txt"
|
28 |
+
# Carga el archivo de texto desde la URL
|
29 |
+
file_content = load_remote_dataset(file_url)
|
30 |
+
# Lee el contenido del archivo y divide por saltos de l铆nea
|
31 |
+
classes = file_content.strip().split("\n")
|
32 |
+
|
33 |
+
|
34 |
+
|
35 |
|
36 |
# Funci贸n para predecir la clase de una nueva imagen
|
37 |
def predict_image(image):
|