sjtans commited on
Commit
13c3232
·
verified ·
1 Parent(s): cba8fb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- model = keras.saving.load_model(model, compile=False)
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