prakasa1234 commited on
Commit
4573ba5
·
verified ·
1 Parent(s): f7673c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ import traceback
10
  # -----------------------------
11
  YOLO_MODEL_PATH = "best.pt"
12
  yolo_model = YOLO(YOLO_MODEL_PATH)
13
- yolo_model.eval()
14
 
15
  # -----------------------------
16
  # 2. Reference alphabet image (WebP)
@@ -29,7 +29,7 @@ def predict_asl(image):
29
 
30
  img = image.copy()
31
 
32
- results = model.predict(str(img_path), imgsz=300, verbose=False)
33
  pred_idx = results[0].probs.top1
34
  pred_label = results[0].names[pred_idx]
35
  confidence = results[0].probs.top1conf.item() * 100
 
10
  # -----------------------------
11
  YOLO_MODEL_PATH = "best.pt"
12
  yolo_model = YOLO(YOLO_MODEL_PATH)
13
+
14
 
15
  # -----------------------------
16
  # 2. Reference alphabet image (WebP)
 
29
 
30
  img = image.copy()
31
 
32
+ results = yolo_model.predict(str(img_path), imgsz=300, verbose=False)
33
  pred_idx = results[0].probs.top1
34
  pred_label = results[0].names[pred_idx]
35
  confidence = results[0].probs.top1conf.item() * 100