Matteo Sirri commited on
Commit
369cd4c
·
1 Parent(s): b0c3cbd

fix: fix description

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -29,7 +29,7 @@ def load_model(baseline: bool = False):
29
 
30
 
31
  def frcnn_motsynth(image):
32
- model = load_model(baseline=True)
33
  transformEval = presets.DetectionPresetEval()
34
  image_tensor = transformEval(image, None)[0]
35
  image_tensor = image_tensor.to(device)
@@ -51,7 +51,7 @@ def frcnn_coco(image):
51
 
52
 
53
  title = "Domain shift adaption on pedestrian detection with Faster R-CNN"
54
- description = "School in AI: Deep Learning, Vision and Language for Industry - second edition final project work by Matteo Sirri"
55
  examples = ["001.jpg", "002.jpg", "003.jpg",
56
  "004.jpg", "005.jpg", "006.jpg", "007.jpg", ]
57
 
@@ -62,4 +62,4 @@ io_custom = gr.Interface(frcnn_motsynth, gr.Image(type="pil"), gr.Image(
62
  type="file", shape=(1920, 1080), label="Faster R-CNN trained on MOTSynth + FT on MOT17"))
63
 
64
  gr.Parallel(io_baseline, io_custom, title=title,
65
- description=description, examples=examples, theme="default").launch(enable_queue=True)
 
29
 
30
 
31
  def frcnn_motsynth(image):
32
+ model = load_model()
33
  transformEval = presets.DetectionPresetEval()
34
  image_tensor = transformEval(image, None)[0]
35
  image_tensor = image_tensor.to(device)
 
51
 
52
 
53
  title = "Domain shift adaption on pedestrian detection with Faster R-CNN"
54
+ description = '<p style="text-align:center">School in AI: Deep Learning, Vision and Language for Industry - second edition final project work by Matteo Sirri.</p> '
55
  examples = ["001.jpg", "002.jpg", "003.jpg",
56
  "004.jpg", "005.jpg", "006.jpg", "007.jpg", ]
57
 
 
62
  type="file", shape=(1920, 1080), label="Faster R-CNN trained on MOTSynth + FT on MOT17"))
63
 
64
  gr.Parallel(io_baseline, io_custom, title=title,
65
+ description=description, examples=examples).launch(enable_queue=True)