Daniel Cerda Escobar commited on
Commit
a16a11f
·
1 Parent(s): d62b128

Update app files

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. utils.py +2 -2
app.py CHANGED
@@ -124,7 +124,7 @@ if submit:
124
  with st.spinner(text="Downloading model weights ... "):
125
  detection_model = get_model()
126
 
127
- image_size = 1024
128
 
129
  with st.spinner(text="Performing prediction ... "):
130
  output = sahi_yolov8m_inference(
@@ -152,7 +152,7 @@ with col2:
152
  img2=st.session_state["output_2"],
153
  label1='Uploaded Diagram',
154
  label2='Model Inference',
155
- width=768,
156
  starting_position=50,
157
  show_labels=True,
158
  make_responsive=True,
 
124
  with st.spinner(text="Downloading model weights ... "):
125
  detection_model = get_model()
126
 
127
+ image_size = 1280
128
 
129
  with st.spinner(text="Performing prediction ... "):
130
  output = sahi_yolov8m_inference(
 
152
  img2=st.session_state["output_2"],
153
  label1='Uploaded Diagram',
154
  label2='Model Inference',
155
+ width=800,
156
  starting_position=50,
157
  show_labels=True,
158
  make_responsive=True,
utils.py CHANGED
@@ -30,8 +30,8 @@ def sahi_yolov8m_inference(
30
  visual_result = sahi.utils.cv.visualize_object_predictions(
31
  image=numpy.array(image),
32
  object_prediction_list=prediction_result.object_prediction_list,
33
- rect_th=2,
34
- text_size=2
35
  )
36
 
37
  output = Image.fromarray(visual_result["image"])
 
30
  visual_result = sahi.utils.cv.visualize_object_predictions(
31
  image=numpy.array(image),
32
  object_prediction_list=prediction_result.object_prediction_list,
33
+ rect_th=3,
34
+ text_size=4
35
  )
36
 
37
  output = Image.fromarray(visual_result["image"])