Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
|
|
| 24 |
|
| 25 |
|
| 26 |
HEADER = """FRAME AI"""
|
| 27 |
-
|
| 28 |
if torch.cuda.is_available():
|
| 29 |
device = "cuda:0"
|
| 30 |
else:
|
|
|
|
| 24 |
|
| 25 |
|
| 26 |
HEADER = """FRAME AI"""
|
| 27 |
+
torch.cuda.empty_cache()
|
| 28 |
if torch.cuda.is_available():
|
| 29 |
device = "cuda:0"
|
| 30 |
else:
|