Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -209,7 +209,7 @@ def generate_images(G, num_images=10): # Reduce for CPU performance
|
|
| 209 |
return z, images
|
| 210 |
|
| 211 |
# Rank images using CLIP
|
| 212 |
-
def rank_by_clip(images, prompt, top_k=3): # Reduce top_k for
|
| 213 |
images_pil = [Image.fromarray(img) for img in images]
|
| 214 |
inputs = clip_processor(text=[prompt], images=images_pil, return_tensors="pt", padding=True)
|
| 215 |
|
|
|
|
| 209 |
return z, images
|
| 210 |
|
| 211 |
# Rank images using CLIP
|
| 212 |
+
def rank_by_clip(images, prompt, top_k=3): # Reduce top_k for speedh
|
| 213 |
images_pil = [Image.fromarray(img) for img in images]
|
| 214 |
inputs = clip_processor(text=[prompt], images=images_pil, return_tensors="pt", padding=True)
|
| 215 |
|