Update modules/pmbl.py
Browse files- modules/pmbl.py +2 -1
modules/pmbl.py
CHANGED
|
@@ -3,7 +3,8 @@ from datetime import datetime
|
|
| 3 |
from concurrent.futures import ThreadPoolExecutor
|
| 4 |
import torch
|
| 5 |
import spaces
|
| 6 |
-
from
|
|
|
|
| 7 |
class PMBL:
|
| 8 |
def __init__(self, model_path):
|
| 9 |
self.model_path = model_path
|
|
|
|
| 3 |
from concurrent.futures import ThreadPoolExecutor
|
| 4 |
import torch
|
| 5 |
import spaces
|
| 6 |
+
from huggingface_hub import spaces
|
| 7 |
+
|
| 8 |
class PMBL:
|
| 9 |
def __init__(self, model_path):
|
| 10 |
self.model_path = model_path
|