sshi commited on
Commit
e209888
1 Parent(s): b3dd9ba

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,7 +100,7 @@ def rescale_bboxes(out_bbox, size):
100
  return b
101
 
102
  def plot_results(pil_img, prob, boxes):
103
- fig = plt.figure(figsize=(64,40), dpi=1200)
104
  plt.imshow(pil_img)
105
  ax = plt.gca()
106
  colors = COLORS * 100
@@ -149,6 +149,6 @@ interface = gr.Interface(
149
  outputs = ['plot'],
150
  examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
151
  title="YOLOS for traffic object detection",
152
- description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a> application on traffic object detection. ")
153
 
154
  interface.launch()
 
100
  return b
101
 
102
  def plot_results(pil_img, prob, boxes):
103
+ fig = plt.figure(figsize=(16,10), dpi=120)
104
  plt.imshow(pil_img)
105
  ax = plt.gca()
106
  colors = COLORS * 100
 
149
  outputs = ['plot'],
150
  examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
151
  title="YOLOS for traffic object detection",
152
+ description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a> which can performe traffic object detection. ")
153
 
154
  interface.launch()