edit model inits

app.py

@@ -18,25 +18,21 @@ def run_models(
         conf_thr:gr.Slider=0.25
 ):
     
-    hp_model = f'{model_paths[model_type]}_hp/best.pt'
-    model = YOLO(hp_model)
-    model.to(device='cpu')
-    hp_result = model(im[:,:,::-1], conf=conf_thr)
+    hp_model = YOLO(f'{model_paths[model_type]}_hp/best.pt')
+    hp_model.to(device='cpu')
+    hp_result = hp_model(im[:,:,::-1], conf=conf_thr)
     hp_im = hp_result[0].plot()
 
-    spk_model = f'{model_paths[model_type]}_spk/best.pt'
-    model = YOLO(spk_model)
-    model.to(device='cpu')
-
-    spk_result = model(im[:,:,::-1], conf=conf_thr)
+    spk_model = YOLO(f'{model_paths[model_type]}_spk/best.pt')
+    spk_model.to(device='cpu')
+    spk_result = spk_model(im[:,:,::-1], conf=conf_thr)
     spk_im = spk_result[0].plot()
 
-    both_model = f'{model_paths[model_type]}_both/best.pt'
-    model = YOLO(both_model)
-    model.to(device='cpu')
-
-    both_result = model(im[:,:,::-1], conf=conf_thr)
+    both_model = YOLO(f'{model_paths[model_type]}_both/best.pt')
+    both_model.to(device='cpu')
+    both_result = both_model(im[:,:,::-1], conf=conf_thr)
     both_im = both_result[0].plot()
+
     return [
         (hp_im[:,:,::-1], 'HP'),
         (spk_im[:,:,::-1], 'SPK'),