Update app/app.py
Browse files- app/app.py +0 -6
app/app.py
CHANGED
|
@@ -55,12 +55,6 @@ class ModelAPI:
|
|
| 55 |
self._setup_routes()
|
| 56 |
|
| 57 |
def _prepare(self):
|
| 58 |
-
|
| 59 |
-
# self.ckpt_path = os.path.join(self.base_path, "dns1680s2.ckpt")
|
| 60 |
-
|
| 61 |
-
# if not os.path.exists(self.ckpt_path):
|
| 62 |
-
# os.makedirs(os.path.dirname(self.ckpt_path), exist_ok=True)
|
| 63 |
-
# from huggingface_hub import hf_hub_download
|
| 64 |
|
| 65 |
self.model = ScoreModel.load_from_checkpoint(self.ckpt_path, self.device)
|
| 66 |
self.model.t_eps = 0.03
|
|
|
|
| 55 |
self._setup_routes()
|
| 56 |
|
| 57 |
def _prepare(self):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
self.model = ScoreModel.load_from_checkpoint(self.ckpt_path, self.device)
|
| 60 |
self.model.t_eps = 0.03
|