Spaces:
Running
on
T4
Running
on
T4
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))
|