Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ if torch.cuda.is_available():
|
|
| 56 |
)
|
| 57 |
|
| 58 |
|
| 59 |
-
@spaces.GPU
|
| 60 |
def get_image_description(image):
|
| 61 |
torch.cuda.empty_cache()
|
| 62 |
gc.collect()
|
|
|
|
| 56 |
)
|
| 57 |
|
| 58 |
|
| 59 |
+
@spaces.GPU
|
| 60 |
def get_image_description(image):
|
| 61 |
torch.cuda.empty_cache()
|
| 62 |
gc.collect()
|