Uppercased

app.py

@@ -300,7 +300,7 @@ 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():
-                device = 'gpu' if torch.cuda.is_available() else 'cpu'
+                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)