sneha
commited on
Commit
•
267367c
1
Parent(s):
8946de5
fix
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ def download_bin():
|
|
| 43 |
os.rename(model_bin, bin_path)
|
| 44 |
|
| 45 |
|
| 46 |
-
def run_attn(input_img,fusion):
|
| 47 |
download_bin()
|
| 48 |
model, embedding_dim, transform, metadata = get_model()
|
| 49 |
if input_img.shape[0] != 3:
|
|
|
|
| 43 |
os.rename(model_bin, bin_path)
|
| 44 |
|
| 45 |
|
| 46 |
+
def run_attn(input_img,fusion="min"):
|
| 47 |
download_bin()
|
| 48 |
model, embedding_dim, transform, metadata = get_model()
|
| 49 |
if input_img.shape[0] != 3:
|