Update: Adjust form

app.py

@@ -173,19 +173,17 @@ def replace_text():
 
 def get_device_info():
     if torch.cuda.is_available():
-        txt = f"Runs on GPU: {torch.cuda.get_device_name(0)}"
+        return f"Runs on GPU: {torch.cuda.get_device_name(0)}"
     else:
-        txt = "Runs on CPU"
-    
-    return f"<span style='color: rgb(113, 113, 122);'>{txt}</span>"
+        return "Runs on CPU (May be slower)"
 
 ### Create Form interface with Gradio Framework ##################################
 with gr.Blocks() as demo:
-    gr.Markdown("## Submit Whisky Review for Classification")
+    gr.HTML("<h2>Submit Whisky Review for Classification</h2>")
 
     with gr.Row():  # alles nebeneinander
         with gr.Column(scale=1):  # linke Seite: Input
-            device_label = gr.Markdown(get_device_info())
+            device_label = gr.HTML(f"<span style='color: var(--block-title-text-color)'>{get_device_info()}</span>")
             review_box = gr.Textbox(
                 label="Whisky Review",
                 lines=8,