[Update] app.py

app.py

@@ -286,8 +286,8 @@ def api():
             streamlit.error('Please upload an image.')
 
 if __name__ == '__main__':
-    print(f"Is CUDA available: {torch.cuda.is_available()}")
-    # True
-    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+    # print(f"Is CUDA available: {torch.cuda.is_available()}")
+    # # True
+    # print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
     # Tesla T4
     api()