scontess commited on
Commit
f3ee846
Β·
1 Parent(s): f84b8cb
Files changed (1) hide show
  1. src/streamlit_app.py +10 -1
src/streamlit_app.py CHANGED
@@ -57,7 +57,16 @@ except PermissionError:
57
 
58
 
59
  # πŸ“Œ Carica il dataset direttamente da Hugging Face
60
- dataset = load_dataset("sradc/imagenet_resized_64x64", split="train")
 
 
 
 
 
 
 
 
 
61
 
62
  #Carica il dataset esterno da imagenet PER TENSORFLOW
63
  #imagenet = tfds.load("imagenet_resized/64x64", split="train", as_supervised=True, download=True, data_dir=DATA_DIR)
 
57
 
58
 
59
  # πŸ“Œ Carica il dataset direttamente da Hugging Face
60
+ dataset = load_dataset("zh-plus/tiny-imagenet", split="train")
61
+
62
+ # πŸ“Œ Recupera il primo esempio
63
+ sample = dataset[0]
64
+ image = sample["image"]
65
+ label = sample["label"]
66
+
67
+ # πŸ“Œ Mostra l'immagine e la classe in Streamlit
68
+ st.image(image, caption=f"Classe: {label}", use_column_width=True)
69
+ st.write(f"πŸ”Ž Esempio dal dataset: {sample}")
70
 
71
  #Carica il dataset esterno da imagenet PER TENSORFLOW
72
  #imagenet = tfds.load("imagenet_resized/64x64", split="train", as_supervised=True, download=True, data_dir=DATA_DIR)