Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def generate_image(text):
|
|
35 |
|
36 |
similarity = text_features.cpu().numpy() @ image_features.cpu().numpy().T
|
37 |
|
38 |
-
|
39 |
index_img = np.argmax(similarity)
|
40 |
img_name = os.path.basename(image_paths[index_img])
|
41 |
im = Image.open(f"./assets/images/{img_name}").convert("RGB")
|
|
|
35 |
|
36 |
similarity = text_features.cpu().numpy() @ image_features.cpu().numpy().T
|
37 |
|
38 |
+
|
39 |
index_img = np.argmax(similarity)
|
40 |
img_name = os.path.basename(image_paths[index_img])
|
41 |
im = Image.open(f"./assets/images/{img_name}").convert("RGB")
|