Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|