Spaces:
Running
Running
Vivien
commited on
Commit
·
7e0130d
1
Parent(s):
81c0cb7
Correct typo
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ if DEBUG:
|
|
| 18 |
OWL_MODEL = f"google/owlvit-base-patch32"
|
| 19 |
else:
|
| 20 |
MODEL = "vit-large-patch14-336"
|
| 21 |
-
OWL_MODEL = f"google/owlvit-large-
|
| 22 |
CLIP_MODEL = f"openai/clip-{MODEL}"
|
| 23 |
|
| 24 |
if not DEBUG and torch.cuda.is_available():
|
|
|
|
| 18 |
OWL_MODEL = f"google/owlvit-base-patch32"
|
| 19 |
else:
|
| 20 |
MODEL = "vit-large-patch14-336"
|
| 21 |
+
OWL_MODEL = f"google/owlvit-large-patch14"
|
| 22 |
CLIP_MODEL = f"openai/clip-{MODEL}"
|
| 23 |
|
| 24 |
if not DEBUG and torch.cuda.is_available():
|