Hugging Face's logo

Spaces:
sasagema
/
prompt-search-engine
Runtime error

App Files Community
sasagema commited on
Commit
4e223f2
·
1 Parent(s): 60da7d9

Change dataset size

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -12,7 +12,7 @@ DATASET = "Gustavosta/Stable-Diffusion-Prompts"
12
 
13
 
14
  model = SentenceTransformer(EMBEDDING_MODEL)
15
- dataset = load_dataset(DATASET , split="test[:10%]")
16
  promptSearchEngine = PromptSearchEngine(dataset["Prompt"], model)
17
 
18
  class SearchRequest(BaseModel):
 
12
 
13
 
14
  model = SentenceTransformer(EMBEDDING_MODEL)
15
+ dataset = load_dataset(DATASET , split="train[:1%]")
16
  promptSearchEngine = PromptSearchEngine(dataset["Prompt"], model)
17
 
18
  class SearchRequest(BaseModel):