Update app.py
Browse files
app.py
CHANGED
|
@@ -221,7 +221,7 @@ def get_ind_to_filter(text, word_ids, keywords):
|
|
| 221 |
|
| 222 |
return inds_to_filter
|
| 223 |
|
| 224 |
-
|
| 225 |
def count(image, text, prompts, state, device):
|
| 226 |
model.to(device)
|
| 227 |
print("state: " + str(state))
|
|
|
|
| 221 |
|
| 222 |
return inds_to_filter
|
| 223 |
|
| 224 |
+
#@spaces.GPU
|
| 225 |
def count(image, text, prompts, state, device):
|
| 226 |
model.to(device)
|
| 227 |
print("state: " + str(state))
|