Jesse-marqo commited on
Commit
42f35f6
·
verified ·
1 Parent(s): 44a0f53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,15 +14,15 @@ def main():
14
  file_path = "merged-averaged-model_timings_2.1.0_12.1_NVIDIA_A10G_False.csv"
15
  # Columns to display
16
  columns_to_display = [
17
- "model_name", "pretrained", "image_time", "text_time",
18
- "avg_score", "image_shape", "text_shape",
19
  "output shape",
20
  "params (M)", "FLOPs (B)"
21
  ] # Specify the columns you want to display
22
 
23
  # Add header and description
24
  st.header("CLIP benchmarks - retrieval and inference")
25
- st.write("CLIP benchmarks for inference and retrieval performance. Image size, context length and output dimensions are also presented. A10G, CUDA 12.1, Torch 2.1.0")
26
 
27
  # Call the display_csv function with the hardcoded file path and selected columns
28
  display_csv(file_path, columns_to_display)
 
14
  file_path = "merged-averaged-model_timings_2.1.0_12.1_NVIDIA_A10G_False.csv"
15
  # Columns to display
16
  columns_to_display = [
17
+ "model_name", "pretrained", "avg_score", "image_time", "text_time",
18
+ "image_shape", "text_shape",
19
  "output shape",
20
  "params (M)", "FLOPs (B)"
21
  ] # Specify the columns you want to display
22
 
23
  # Add header and description
24
  st.header("CLIP benchmarks - retrieval and inference")
25
+ st.write("CLIP benchmarks for inference and retrieval performance. Image size, context length and output dimensions are also included. Retrieval performance comes from https://github.com/mlfoundations/open_clip/blob/main/docs/openclip_retrieval_results.csv.\n A10G, CUDA 12.1, Torch 2.1.0")
26
 
27
  # Call the display_csv function with the hardcoded file path and selected columns
28
  display_csv(file_path, columns_to_display)