Update app.py

app.py

@@ -20,38 +20,16 @@ def generate_image_from_prompt(prompt_text):
 title = "KVIImager 2.0 Demo 🎨"
 description = "This app uses Hugging Face AI model to generate an image based on the provided text prompt 🖼️."
 
-css_styles = {
-    'body': {
-        'background-color': '#f4f4f4',
-        'font-family': 'Arial, sans-serif'
-    },
-    'title': {
-        'color': 'navy',
-        'font-size': '36px',
-        'text-align': 'center',
-        'margin-bottom': '20px'
-    },
-    'textbox': {
-        'border': '2px solid #008CBA',
-        'border-radius': '5px',
-        'padding': '10px',
-        'margin-bottom': '20px',
-        'width': '300px',
-        'font-size': '16px'
-    },
-    'output_image': {
-        'box-shadow': '2px 2px 5px #888888'
-    }
-}
+input_prompt = gr.inputs.Textbox(label="Enter Prompt 📝", placeholder="E.g. 'A peaceful garden with a small cottage'")
+output_generated_image = gr.outputs.Image(label="Generated Image")
 
-input_prompt = gr.Textbox(label="Enter Prompt 📝", placeholder="E.g. 'A peaceful garden with a small cottage'")
-output_generated_image = gr.Image(label="Generated Image")
-
-gr.Interface(
-    fn=generate_image_from_prompt, 
-    inputs=input_prompt, 
-    outputs=output_generated_image, 
-    title=title,
-    description=description,
-    #examples=examples
-).launch(inline=True, css=css_styles)
+with gr.Blocks(theme=gr.theme.Soft()) as app:
+    caption = "Generate Image"
+    iface = gr.Interface(
+        generate_image_from_prompt,
+        inputs=input_prompt, 
+        outputs=output_generated_image, 
+        title=title,
+        description=description
+    )
+    iface.launch(share=True, inline=True)