Compute time display fix

app.py

@@ -5,6 +5,7 @@ from io import StringIO
 import pdb
 import json
 from twc_embeddings import HFModel,SimCSEModel,SGPTModel
+import torch
 
 
 MAX_INPUT = 100
@@ -151,7 +152,7 @@ def construct_model_info_for_display():
         if (node["mark"] == True):
             markdown_str += f"<div style=\"font-size:16px; color: #5f5f5f; text-align: left\">&nbsp;•&nbsp;Model:&nbsp;<a href=\'{node['paper_url']}\' target='_blank'>{node['name']}</a><br/>&nbsp;&nbsp;&nbsp;&nbsp;Code released by:&nbsp;<a href=\'{node['orig_author_url']}\' target='_blank'>{node['orig_author']}</a><br/>&nbsp;&nbsp;&nbsp;&nbsp;Model info:&nbsp;<a href=\'{node['sota_info']['sota_link']}\' target='_blank'>{node['sota_info']['task']}</a></div>"
             if ("Note" in node):
-                markdown_str += f"<div style=\"font-size:16px; color: #dd1515; text-align: left\">&nbsp;&nbsp;&nbsp;&nbsp;{node['Note']}<a href=\'{node['alt_url']}\' target='_blank'>link</a></div>"
+                markdown_str += f"<div style=\"font-size:16px; color: #a91212; text-align: left\">&nbsp;&nbsp;&nbsp;&nbsp;{node['Note']}<a href=\'{node['alt_url']}\' target='_blank'>link</a></div>"
             markdown_str += "<div style=\"font-size:16px; color: #5f5f5f; text-align: left\"><br/></div>"
         
     markdown_str += "<div style=\"font-size:12px; color: #9f9f9f; text-align: left\"><b>Note:</b><br/>•&nbsp;Uploaded files are loaded into non-persistent memory for the duration of the computation. They are not saved</div>"
@@ -299,7 +300,8 @@ def main():
             results = run_test(selected_model,sentences,display_area,main_index - 1,(uploaded_file is not None))
             display_area.empty()
             with display_area.container():
-                response_info = f"Response time - {time.time() - start:.2f} secs for {len(sentences)} sentences"
+                device = 'gpu' if torch.cuda.is_available() else 'cpu'
+                response_info = f"Computation time on {device}: {time.time() - start:.2f} secs for {len(sentences)} sentences"
                 display_results(sentences,main_index - 1,results,response_info)
                 #st.json(results)
       st.download_button(