Update app.py

app.py

@@ -1,6 +1,15 @@
 import time
 import logging
 import gradio as gr
+import gradio as gr
+
+theme = gr.themes.Glass(
+    primary_hue="amber",
+    secondary_hue="orange",
+    neutral_hue="amber",
+)
+
+with gr.Blocks(theme=theme) as demo:
 
 from src.llm_boilers import llm_boiler
 
@@ -110,15 +119,6 @@ def call_inf_server(prompt, openai_key):
         logging.warning(f"Result of text generation: {response}")
         return response
 
-# ------ STYLES --- 
-#  see https://huggingface.co/spaces/Gradio-Themes/theme_builder for more to tweek
-        #//////////////////////////////////////////////////////////////////// Glass(),
-        
-with gr.Blocks(theme = gr.themes.Glass(
-    primary_hue="amber",
-    secondary_hue="orange",
-    neutral_hue="amber",
-) ):
 
 with gr.Blocks(theme=theme) as demo:
      gr.Markdown(