Update app.py

app.py

@@ -1,4 +1,4 @@
-import os
+
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import spaces
@@ -64,20 +64,9 @@ def generate_kids_story(character, setting, language):
     gc.collect ()
     # Empty CUDA cache
     torch.cuda.empty_cache()
-
     
     return output[0]["generated_text"]
 
-css_style = """
-    body {
-        background-image: url('https://cdna.artstation.com/p/assets/images/images/074/776/904/large/pietro-chiovaro-r1-castle-chp.jpg?1712916847');
-        background-size: cover;
-        background-position: center;
-        color: #fff;  /* General text color */
-        font-family: 'Arial', sans-serif;
-    }
-"""
-
 # Create Gradio interface
 demo = gr.Interface(
     fn=generate_kids_story,
@@ -99,7 +88,7 @@ demo = gr.Interface(
         ["Lila the Ladybug", "a garden full of flowers", "English"],
         ["ليلى الجنية", "حديقة مليئة بالأزهار", "Arabic"]
     ],
-    css = css_style
+    theme = de,fault,
 )
 
 # Launch the Gradio app