Update app.py

app.py

@@ -8,23 +8,19 @@ from PIL import Image
 
 # Configurações do dispositivo para uso apenas da CPU
 device = "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Continuando com o modelo especificado
+model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"  # Continuando com o modelo especificado
 
 # Carregar o pipeline configurado para CPU
 pipe = DiffusionPipeline.from_pretrained(model_repo_id)
 pipe = pipe.to(device)
 
 # Carregando o dataset do Hugging Face
-dataset = load_dataset("LEIDIA/Data_Womleimg")
-
-
-# Carrega o dataset de imagens
-dataset = load_dataset("LEIDIA/Data_Womleimg", split="train")
+dataset = load_dataset("LEIDIA/Data_Womleimg", split="train")# Carrega o dataset de imagens
 
 # Acessando as descrições
 for example in dataset:
     image_id = example["image"].filename
-    description = example["descricoes"]  # Acessando a nova feature
+    description = example["Wom_dataset"]  # Acessando a nova feature
     print(f"ID da imagem: {image_id}, Descrição: {description}")
 
     # Cria um dicionário para mapear IDs de imagem às descrições