sahadev10 commited on
Commit
47a7b61
·
verified ·
1 Parent(s): d9c7a02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 speed
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