Update app.py

app.py

@@ -4,6 +4,13 @@ from urllib.request import urlopen
 import json
 import time
 from load_llms import model_choices, llm_intro, load_model
+from huggingface_hub import HfApi
+
+
+repo_id = "yxmauw/chatbot"
+api = HfApi()
+runtime = api.get_space_runtime(repo_id=repo_id)
+
 
     
 # Construct chatbot
@@ -38,14 +45,16 @@ with gr.Blocks(
             # Link the dropdown with the textbox to update the description based on the selected model
             model_dropdown.change(fn=llm_intro, inputs=model_dropdown, outputs=explanation)
 
-            gr.Gallery(label="Current Space Hardware used",             
-                       value=["space_hardware.png"], 
-                       type="filepath", 
-                       columns=1,
-                       object_fit="scale-down",
-                       height=100,
-                       container=False
-                      )
+            gr.Markdown(f"## Running on {runtime.hardware}")
+
+            # gr.Gallery(label="Current Space Hardware used",             
+            #            value=["space_hardware.png"], 
+            #            type="filepath", 
+            #            columns=1,
+            #            object_fit="scale-down",
+            #            height=100,
+            #            container=False
+            #           )
 
         with gr.Column(scale=4, elem_id='col'):
             chatbot = gr.Chatbot(label="Chatroom", value=[(None, "How may I help you today?")], elem_id="chatbot")