farrell236 commited on
Commit
ee37cdf
·
1 Parent(s): 20b03eb
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -107,16 +107,13 @@ def format_probs(probabilities):
107
 
108
 
109
  def model_info(args, device):
110
- return f"""
111
- # DeepSeeNet
112
-
113
- <div style="display: grid; grid-template-columns: repeat(4, max-content); gap: 0.75rem 2rem; align-items: center;">
114
- <div><b>Model</b><br><code>{args.backbone}</code></div>
115
- <div><b>Input size</b><br><code>{args.image_size} × {args.image_size}</code></div>
116
- <div><b>Device</b><br><code>{device.type}</code></div>
117
- <div><b>Checkpoint folder</b><br><code>{args.checkpoint_folder}</code></div>
118
- </div>
119
- """
120
 
121
 
122
  def make_app(args):
 
107
 
108
 
109
  def model_info(args, device):
110
+ return (
111
+ "# DeepSeeNet\n\n"
112
+ f"Model: `{args.backbone}` \n"
113
+ f"Input size: `{args.image_size} × {args.image_size}` \n"
114
+ f"Device: `{device.type}` \n"
115
+ f"Checkpoint folder: `{args.checkpoint_folder}`"
116
+ )
 
 
 
117
 
118
 
119
  def make_app(args):