SeptAlfauzan commited on
Commit
12cee8d
1 Parent(s): 713f67e

update: change model path for production

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. requirements.txt +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from PIL import Image
3
  import torch
4
  from ultralyticsplus import YOLO, render_result
 
5
 
6
 
7
  def launch(
@@ -12,9 +13,7 @@ def launch(
12
  ):
13
  try:
14
  model_path = "./models/student-behaviour-best.pt"
15
- model = YOLO(
16
- "./student-behaviour-test-deploy/models/OWN-DATASET-640-e120-b32-best.pt"
17
- )
18
 
19
  # pil_image = Image.fromarray(image)
20
 
 
2
  from PIL import Image
3
  import torch
4
  from ultralyticsplus import YOLO, render_result
5
+ from path import path
6
 
7
 
8
  def launch(
 
13
  ):
14
  try:
15
  model_path = "./models/student-behaviour-best.pt"
16
+ model = YOLO("./models/OWN-DATASET-640-e120-b32-best.pt")
 
 
17
 
18
  # pil_image = Image.fromarray(image)
19
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  gradio==4.24.0
2
  torch==2.2.1
3
  ultralyticsplus==0.1.0
 
 
1
  gradio==4.24.0
2
  torch==2.2.1
3
  ultralyticsplus==0.1.0
4
+ path