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
|