Update app.py

app.py

@@ -77,7 +77,7 @@ def main():
 
   device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
   
-  st.info('Using device type: {}'.format(device))
+  
   
   embeddings = load_prep_data()
 
@@ -92,6 +92,8 @@ def main():
   
   query = st.text_input("Enter Query",'What are the corona viruses?', key="query")
   
+  st.write('Using device type: {}'.format(device))
+  
   with st.spinner('Please wait...'):
     dt1 = datetime.datetime.now()