Update torch device information

app.py

@@ -245,6 +245,8 @@ def dataset_load():
         st.write(f'Product Name : {prod_name}')
         st.write(f'Product ID   : {selected_product_id} ')
         st.write(f'Total Rows   : {preview_df.shape[0]}')
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+        st.write(f"Selected device for processing is (CPU/GPU) : {device.upper()}")
     
     #================================================================
     # Use the condition to control the display of the radio buttons
@@ -280,9 +282,9 @@ def dataset_load():
         # =========================================================================================================================
         st.markdown("---")
         st.subheader("Step 3 : Initialising the models & running operation")
-        with st.spinner("Initializing RoBERTa Model ............"):
-            device = "cuda" if torch.cuda.is_available() else "cpu"
-            st.write(f"Selected device for processing is (CPU/GPU) : {device.upper()}")
+        #with st.spinner("Initializing RoBERTa Model ............"):
+            #device = "cuda" if torch.cuda.is_available() else "cpu"
+            #st.write(f"Selected device for processing is (CPU/GPU) : {device.upper()}")
 
         # ROBERTA Model
         with st.spinner("Initializing RoBERTa Model ............"):