Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 import spaces
 
-
 def get_model_name(language):
     """Map language choice to the corresponding model."""
     model_mapping = {
@@ -30,6 +29,7 @@ def load_model(model_name):
         do_sample=False
     )
     return generator
+
 @spaces.GPU
 def generate_plant_info(plant_name, language):
     model_name = get_model_name(language)
@@ -49,16 +49,15 @@ def generate_plant_info(plant_name, language):
     output = generator(messages)
     return output[0]["generated_text"]
 
-
-# Create Gradio interface
+# Create Gradio interface with enhancements
 demo = gr.Interface(
     fn=generate_plant_info,
     inputs=[
         gr.Textbox(placeholder="Enter plant name (e.g., Lavender, Aloe Vera)...", label="Plant Name"),
         gr.Dropdown(
-            choices=["English", "Arabic"],  # Users choose the language, not the model name
+            choices=["English", "Arabic"],
             label="Choose Language",
-            value="English"  # Default to English
+            value="English"
         )
     ],
     outputs=gr.Textbox(label="Plant Information"),
@@ -71,6 +70,29 @@ demo = gr.Interface(
         ["الصبار", "Arabic"]
     ],
     theme="default",
+    layout="vertical",  # Change layout for better organization
 )
 
-demo.launch()
+# Custom CSS for styling
+custom_css = """
+<style>
+    body {
+        font-family: Arial, sans-serif;
+        background-color: #f9f9f9;
+    }
+    .gri-container {
+        border-radius: 10px;
+        background-color: #fff;
+        box-shadow: 0 4px 20px rgba(0,0,0,0.1);
+    }
+    .gri-button {
+        background-color: #4CAF50;
+        color: white;
+    }
+    .gri-button:hover {
+        background-color: #45a049;
+    }
+</style>
+"""
+
+demo.launch(share=True, css=custom_css)  # Launch the interface with custom CSS