Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -362,6 +362,7 @@ def add_boxes_norm_path(filepath, boxes):
|
|
| 362 |
|
| 363 |
@spaces.GPU(duration=60)
|
| 364 |
def run_model_gpu60(model, tensor):
|
|
|
|
| 365 |
result = model(tensor).detach().numpy()
|
| 366 |
return result
|
| 367 |
|
|
@@ -382,7 +383,7 @@ def detect_cells(filepath, c, model, rescale_z, rescale_xy, progress=gr.Progress
|
|
| 382 |
model = download_model(model_adresses[model])
|
| 383 |
#model = tf.keras.models.load_model(model_adresses[model], compile=False)
|
| 384 |
# model = keras.saving.load_model("hf://sjtans/OrganoidTracker2_pytorch", compile=False)
|
| 385 |
-
|
| 386 |
|
| 387 |
img = get_volume(filepath[-1], c = c)
|
| 388 |
|
|
|
|
| 362 |
|
| 363 |
@spaces.GPU(duration=60)
|
| 364 |
def run_model_gpu60(model, tensor):
|
| 365 |
+
model = keras.saving.load_model(model, compile=False)
|
| 366 |
result = model(tensor).detach().numpy()
|
| 367 |
return result
|
| 368 |
|
|
|
|
| 383 |
model = download_model(model_adresses[model])
|
| 384 |
#model = tf.keras.models.load_model(model_adresses[model], compile=False)
|
| 385 |
# model = keras.saving.load_model("hf://sjtans/OrganoidTracker2_pytorch", compile=False)
|
| 386 |
+
# model = keras.saving.load_model(model, compile=False)
|
| 387 |
|
| 388 |
img = get_volume(filepath[-1], c = c)
|
| 389 |
|