Fixed the bugs

app.py

@@ -1,69 +1,119 @@
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+
+MODEL_ID = "GinieAI/Solidity-LLM"
+
 print("Loading Ginie Solidity LLM...")
-tokenizer = AutoTokenizer.from_pretrained("GinieAI/Solidity-LLM")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model = AutoModelForCausalLM.from_pretrained(
-    "GinieAI/Solidity-LLM",
-    torch_dtype=torch.float32,   # CPU-safe for Space free tier
+    MODEL_ID,
+    torch_dtype=torch.float32,
+    low_cpu_mem_usage=True,
     device_map="cpu"
 )
+model.eval()
 print("Model ready!")
+
+
 def generate_contract(instruction, max_tokens=600, temperature=0.7):
-    prompt = f"""### Instruction:
-{instruction}
-### Response:
-"""
-    inputs = tokenizer(prompt, return_tensors="pt")
+    if not instruction.strip():
+        return "Please enter a contract description."
+
+    prompt = f"### Instruction:\n{instruction.strip()}\n\n### Response:\n"
+
+    inputs = tokenizer(
+        prompt,
+        return_tensors="pt",
+        truncation=True,
+        max_length=512
+    )
+
     with torch.no_grad():
         outputs = model.generate(
-            **inputs,
-            max_new_tokens=max_tokens,
-            temperature=temperature,
-            do_sample=True,
-            pad_token_id=tokenizer.eos_token_id
+            input_ids=inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],
+            max_new_tokens=int(max_tokens),
+            temperature=float(temperature),
+            do_sample=temperature > 0,
+            pad_token_id=tokenizer.eos_token_id,
+            eos_token_id=tokenizer.eos_token_id,
         )
-    result = tokenizer.decode(outputs, skip_special_tokens=True)
-    # Return only the response part
-    if "### Response:" in result:
-        result = result.split("### Response:")[-1].strip()
-    return result
+
+    generated = tokenizer.decode(
+        outputs[0][inputs["input_ids"].shape[1]:],
+        skip_special_tokens=True
+    ).strip()
+
+    return generated
+
+
 examples = [
-    ["Write a Solidity ERC20 token contract with minting and burning."],
+    ["Write a Solidity ERC20 token with minting, burning, and owner controls."],
     ["Write a Solidity multisig wallet requiring 2 of 3 signatures."],
     ["Write a Solidity staking contract with 10% APY rewards."],
-    ["Write a Solidity DAO governance contract with voting."],
-    ["Write a Solidity NFT contract with whitelist minting."],
+    ["Write a Solidity DAO governance contract with proposal voting."],
+    ["Write a Solidity NFT contract with whitelist minting and reveal."],
 ]
-with gr.Blocks(theme=gr.themes.Soft(), title="Ginie AI") as demo:
+
+css = """
+.output-code { font-family: monospace; font-size: 13px; }
+footer { display: none !important; }
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(), title="Ginie AI — Smart Contract Generator", css=css) as demo:
+
     gr.Markdown("""
-    # 🧞 Ginie AI — Solidity Smart Contract Generator
-    **The AI built for Web3 developers** · [ginie.xyz](https://ginie.xyz)
+# Ginie AI — Solidity Smart Contract Generator
 
-    Describe the smart contract you need in plain English. Ginie generates production-ready Solidity.
+The AI built for Web3 developers · [ginie.xyz](https://ginie.xyz) · [npm SDK](https://npmjs.com/package/ginie-sdk) — 30k+ weekly downloads
+
+Describe the contract you need in plain English. Ginie generates production-ready Solidity.
     """)
 
     with gr.Row():
         with gr.Column(scale=1):
             instruction = gr.Textbox(
-                label="What contract do you need?",
+                label="Describe your contract",
                 placeholder="e.g. Write a Solidity ERC20 token with minting, burning, and owner controls.",
-                lines=4
+                lines=5
             )
             with gr.Row():
-                max_tokens = gr.Slider(100, 1000, value=600, label="Max tokens")
-                temperature = gr.Slider(0.1, 1.0, value=0.7, label="Creativity")
-            btn = gr.Button("🧞 Generate Contract", variant="primary")
+                max_tokens = gr.Slider(
+                    minimum=100, maximum=800, value=600, step=50,
+                    label="Max tokens"
+                )
+                temperature = gr.Slider(
+                    minimum=0.1, maximum=1.0, value=0.7, step=0.1,
+                    label="Temperature"
+                )
+            btn = gr.Button("Generate Contract", variant="primary", size="lg")
 
         with gr.Column(scale=2):
-            output = gr.Code(label="Generated Solidity", language="javascript")
+            output = gr.Code(
+                label="Generated Solidity",
+                language="javascript",
+                lines=30,
+                elem_classes=["output-code"]
+            )
+
+    gr.Examples(
+        examples=examples,
+        inputs=instruction,
+        label="Example prompts"
+    )
 
-    gr.Examples(examples=examples, inputs=instruction)
-    btn.click(generate_contract, inputs=[instruction, max_tokens, temperature], outputs=output)
+    btn.click(
+        fn=generate_contract,
+        inputs=[instruction, max_tokens, temperature],
+        outputs=output
+    )
 
     gr.Markdown("""
-    ---
-    **⚠️ Always review AI-generated code before deploying to mainnet.**
-    Model: [GinieAI/Solidity-LLM](https://huggingface.co/GinieAI/Solidity-LLM) · License: MIT
+---
+Model: [GinieAI/Solidity-LLM](https://huggingface.co/GinieAI/Solidity-LLM) · Base: Chain-GPT/Solidity-LLM · License: MIT
+
+> Always review AI-generated contracts before deploying to mainnet.
     """)
-demo.launch()
+
+demo.launch()