Update app.py

app.py

@@ -392,6 +392,7 @@ model = AutoModelForCausalLM.from_pretrained(
 
 import gradio as gr
 import torch
+@spaces.GPU
 def generate_from_prompt(prompt):
     try:
         messages = [{"role": "user", "content": prompt}]
@@ -418,6 +419,7 @@ def generate_from_prompt(prompt):
 
     except Exception as e:
         return f"Error during generation: {e}"
+@spaces.GPU
 def generate_model_from_description(description):
     if not description.strip():
         return "Error: Description is required."
@@ -560,8 +562,27 @@ with gr.Blocks(title="C# Validator Generator with 8 Tabs") as demo:
                 outputs=[model_code_output_desc]
             )
 
+        with gr.Tab("Generate  from Description"):
+            with gr.Row():
+                with gr.Column():
+                    description_inputt = gr.Textbox(
+                        label="Project Description",
+                        lines=10,
+                        placeholder="Enter your project or model description here..."
+                    )
+                    generate_model_desc_btnt = gr.Button("Generate Model Code")
+        
+                with gr.Column():
+                    model_code_output_desct = gr.Code(label="Generated Model Code", lines=25)
+        
+            generate_model_desc_btnt.click(
+                fn=generate_from_prompt,
+                inputs=[description_inputt],
+                outputs=[model_code_output_desct]
+            )
+
         # Tab 3 to Tab 8 placeholders for future features
-        for i in range(4, 9):
+        for i in range(5, 9):
             with gr.Tab(f"Tab {i} - Placeholder"):
                 gr.Markdown(f"Content for Tab {i} goes here...")