theme

app.py

@@ -68,6 +68,7 @@ def florence_caption(image):
     return parsed_answer["<MORE_DETAILED_CAPTION>"]
 
 # Prompt Enhancer function
+@spaces.GPU
 def enhance_prompt(input_prompt):
     result = enhancer_long("Enhance the description: " + input_prompt)
     enhanced_text = result[0]['summary_text']
@@ -127,24 +128,7 @@ def generate_llm_prompt(input_text, provider, model, prompt_type):
     except Exception as e:
         print(f"An error occurred in generate_llm_prompt: {e}")
         return input_text  # Return original input if there's an error
-
-custom_css = """
-.input-group, .output-group {
-    border: 1px solid #e0e0e0;
-    border-radius: 10px;
-    padding: 20px;
-    margin-bottom: 20px;
-    background-color: #f9f9f9;
-}
-.submit-btn {
-    background-color: #2980b9 !important;
-    color: white !important;
-}
-.submit-btn:hover {
-    background-color: #3498db !important;
-}
-"""
-
+    
 title = """<h1 align="center">Stable Diffusion 3.5 with Florence-2 Captioner and Prompt Enhancer</h1>
 <p><center>
 <a href="https://huggingface.co/stabilityai/stable-diffusion-3.5-large" target="_blank">[Stable Diffusion 3.5 Model]</a>
@@ -154,7 +138,7 @@ title = """<h1 align="center">Stable Diffusion 3.5 with Florence-2 Captioner and
 </center></p>
 """
 
-with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")) as demo:
+with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     gr.HTML(title)
     
     with gr.Row():