Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -305,7 +305,6 @@ def process(query_img, node_descriptions, edges,
|
|
| 305 |
|
| 306 |
|
| 307 |
#model.cuda()
|
| 308 |
-
model.to('cuda')
|
| 309 |
model.eval()
|
| 310 |
# return model(**data)
|
| 311 |
# with torch.no_grad():
|
|
|
|
| 305 |
|
| 306 |
|
| 307 |
#model.cuda()
|
|
|
|
| 308 |
model.eval()
|
| 309 |
# return model(**data)
|
| 310 |
# with torch.no_grad():
|