LEIDIA commited on
Commit
3a6f3d0
·
verified ·
1 Parent(s): b27eba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -8,23 +8,19 @@ from PIL import Image
8
 
9
  # Configurações do dispositivo para uso apenas da CPU
10
  device = "cpu"
11
- model_repo_id = "stabilityai/sdxl-turbo" # Continuando com o modelo especificado
12
 
13
  # Carregar o pipeline configurado para CPU
14
  pipe = DiffusionPipeline.from_pretrained(model_repo_id)
15
  pipe = pipe.to(device)
16
 
17
  # Carregando o dataset do Hugging Face
18
- dataset = load_dataset("LEIDIA/Data_Womleimg")
19
-
20
-
21
- # Carrega o dataset de imagens
22
- dataset = load_dataset("LEIDIA/Data_Womleimg", split="train")
23
 
24
  # Acessando as descrições
25
  for example in dataset:
26
  image_id = example["image"].filename
27
- description = example["descricoes"] # Acessando a nova feature
28
  print(f"ID da imagem: {image_id}, Descrição: {description}")
29
 
30
  # Cria um dicionário para mapear IDs de imagem às descrições
 
8
 
9
  # Configurações do dispositivo para uso apenas da CPU
10
  device = "cpu"
11
+ model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0" # Continuando com o modelo especificado
12
 
13
  # Carregar o pipeline configurado para CPU
14
  pipe = DiffusionPipeline.from_pretrained(model_repo_id)
15
  pipe = pipe.to(device)
16
 
17
  # Carregando o dataset do Hugging Face
18
+ dataset = load_dataset("LEIDIA/Data_Womleimg", split="train")# Carrega o dataset de imagens
 
 
 
 
19
 
20
  # Acessando as descrições
21
  for example in dataset:
22
  image_id = example["image"].filename
23
+ description = example["Wom_dataset"] # Acessando a nova feature
24
  print(f"ID da imagem: {image_id}, Descrição: {description}")
25
 
26
  # Cria um dicionário para mapear IDs de imagem às descrições