Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,8 +52,6 @@ device_map = infer_auto_device_map(
|
|
| 52 |
|
| 53 |
model = dispatch_model(model, device_map=device_map)
|
| 54 |
|
| 55 |
-
device = ‘cuda:0’
|
| 56 |
-
|
| 57 |
class calculateDuration:
|
| 58 |
def __init__(self, activity_name=""):
|
| 59 |
self.activity_name = activity_name
|
|
|
|
| 52 |
|
| 53 |
model = dispatch_model(model, device_map=device_map)
|
| 54 |
|
|
|
|
|
|
|
| 55 |
class calculateDuration:
|
| 56 |
def __init__(self, activity_name=""):
|
| 57 |
self.activity_name = activity_name
|