Add GPUs

app.py

@@ -2,22 +2,21 @@ import streamlit as st
 
 st.title("GCP Resource Alloctor")
 
-st.header("GCP Resource Alloctor")
 
-# # GPU Type
-# gpu_type = st.selectbox(
-#     'Which type of GPU you would like to use?',
-#     (  
-#         'NVIDIA H100 80GB', 
-#         'NVIDIA A100 80GB',
-#         'NVIDIA A100 40GB',
-#         'NVIDIA V100 16GB',
-#         'NVIDIA P100 16GB',
-#         'NVIDIA L4 24GB',
-#         'NVIDIA T4 16GB',
-#         'NVIDIA P4 8GB',
-#     )
-# )
+# GPU Type
+gpu_type = st.selectbox(
+    'Which type of GPU you would like to use?',
+    (  
+        'NVIDIA H100 80GB', 
+        'NVIDIA A100 80GB',
+        'NVIDIA A100 40GB',
+        'NVIDIA V100 16GB',
+        'NVIDIA P100 16GB',
+        'NVIDIA L4 24GB',
+        'NVIDIA T4 16GB',
+        'NVIDIA P4 8GB',
+    )
+)
 
 # # GPU Numbers
 # gpu_numbers = st.slider('How many GPUs you would like to use?', 0, 8, 1)