Alan7 commited on
Commit
b563d2d
verified
1 Parent(s): 1fe5148

modifico la lectura de clases

Browse files
Files changed (1) hide show
  1. app.py +14 -3
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):