linoyts HF Staff commited on
Commit
152624c
·
verified ·
1 Parent(s): ee8f1f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -223,6 +223,7 @@ def process_images(
223
  def get_text_embeddings(concept_file):
224
  descriptions = load_descriptions(concept_file)
225
  embeddings = generate_embeddings(descriptions, clip_model, tokenizer, device, batch_size=100)
 
226
  return embeddings, True
227
 
228
 
@@ -255,7 +256,13 @@ def process_and_display(
255
  concept_image2, concept_name2,
256
  concept_image3, concept_name3,
257
  rank1, rank2, rank3,
258
- prompt, scale, seed, num_inference_steps
 
 
 
 
 
 
259
  )
260
 
261
  return modified_images
 
223
  def get_text_embeddings(concept_file):
224
  descriptions = load_descriptions(concept_file)
225
  embeddings = generate_embeddings(descriptions, clip_model, tokenizer, device, batch_size=100)
226
+ print("done with text embeddings")
227
  return embeddings, True
228
 
229
 
 
256
  concept_image2, concept_name2,
257
  concept_image3, concept_name3,
258
  rank1, rank2, rank3,
259
+ prompt, scale, seed, num_inference_steps,
260
+ concpet_from_file_1,
261
+ concpet_from_file_2,
262
+ concpet_from_file_3,
263
+ use_concpet_from_file_1,
264
+ use_concpet_from_file_2,
265
+ use_concpet_from_file_3
266
  )
267
 
268
  return modified_images