Spaces:

jeffyuyu
/

text-to-image

Sleeping

jeffyuyu commited on Apr 9

Commit

4fae3d7

verified ·

1 Parent(s): 2e59db3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ MAX_IMAGE_SIZE = 1024
 # 載入你的 dataset，這裡假設有一個 "prompt" 欄位
 dataset = load_dataset("jeffyuyu/labeled_images_demo_BLIP2", split="train")
-dataset_prompts = list(dataset["prompt"])
 def infer(
     prompt,

 # 載入你的 dataset，這裡假設有一個 "prompt" 欄位
 dataset = load_dataset("jeffyuyu/labeled_images_demo_BLIP2", split="train")
+dataset_prompts = list(dataset["label"])
 def infer(
     prompt,