Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
app.py
CHANGED
@@ -256,6 +256,9 @@ hands = mp_hands.Hands(
|
|
256 |
|
257 |
@spaces.GPU(duration=120)
|
258 |
def get_ref_anno(ref):
|
|
|
|
|
|
|
259 |
if ref is None:
|
260 |
return (
|
261 |
None,
|
|
|
256 |
|
257 |
@spaces.GPU(duration=120)
|
258 |
def get_ref_anno(ref):
|
259 |
+
print(torch.cuda.is_available())
|
260 |
+
print(torch.cuda.device_count())
|
261 |
+
print(torch.cuda.get_device_name(0))
|
262 |
if ref is None:
|
263 |
return (
|
264 |
None,
|