Update app.py

app.py

@@ -4,7 +4,6 @@ import random
 import spaces
 import torch
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
-from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -40,17 +39,20 @@ examples = [
 
 css = """
 body {
-    background-color: #e0f7fa;
+    background-color: #f4faff;
     color: #005662;
+    font-family: Arial, sans-serif;
 }
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 100%;
+    padding: 20px;
 }
 .gr-button {
     background-color: #0288d1;
     color: white;
     border-radius: 8px;
+    transition: background-color 0.3s ease;
 }
 .gr-button:hover {
     background-color: #0277bd;
@@ -75,6 +77,11 @@ body {
 .gr-slider-thumb {
     background-color: #005662 !important;
 }
+.gr-text-input, .gr-image {
+    width: 100%;
+    box-sizing: border-box;
+    margin-bottom: 10px;
+}
 """
 
 with gr.Blocks(css=css) as demo: