Update app.py

app.py

@@ -4,12 +4,13 @@ from pydantic import BaseModel
 import torch
 import re
 import secrets
-import requests # Google Search API එකට අවශ්‍යයි
+import requests
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-app = FastAPI()
+# Hugging Face සර්වර් එක 'main' ලෝඩ් කිරීමට උත්සාහ කරන බැවින් මෙය මෙසේ නම් කරන ලදී.
+main = FastAPI()
 
-app.add_middleware(
+main.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_methods=["*"],
@@ -24,18 +25,20 @@ API_KEYS_DB = {
 ADMIN_SECRET = "MINZO-SECRET-2026"
 
 # ── Google Search Config ──
+# MINZO-PRIME, ඔයාගේ Keys මෙතනට ඇතුළත් කරන්න.
 GOOGLE_API_KEY = "YOUR_GOOGLE_API_KEY"
 GOOGLE_CX = "YOUR_CUSTOM_SEARCH_ENGINE_ID"
 
 # ── Load AI Model for CPU ──
-model_id = "google/gemma-4-E4B-it" 
+model_id = "google/gemma-2-9b-it" 
 print(f"Loading {model_id} on CPU (Optimized for 18GB RAM)...")
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-# CPU එකේ දුවන නිසා torch_dtype එක float16 හෝ bfloat16 දැමීමෙන් RAM එක ඉතිරි කරගත හැක
+
+# 'torch_dtype' වෙනුවට අලුත් 'dtype' පරාමිතිය භාවිතා කර ඇත.
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
-    torch_dtype=torch.bfloat16, 
+    dtype=torch.bfloat16, 
     device_map="cpu", 
     trust_remote_code=True
 )
@@ -96,7 +99,7 @@ def should_search(query: str) -> bool:
 # ──────────────────────────────────────
 # ENDPOINTS
 # ──────────────────────────────────────
-@app.get("/")
+@main.get("/")
 def home():
     return {
         "status": "Inachi-Core Active",
@@ -104,7 +107,7 @@ def home():
         "search": "Google Real-time",
     }
 
-@app.post("/v1/generate-key")
+@main.post("/v1/generate-key")
 async def generate_key(data: AdminRequest):
     if data.admin_pass != ADMIN_SECRET:
         raise HTTPException(status_code=401, detail="Unauthorized Access!")
@@ -112,7 +115,7 @@ async def generate_key(data: AdminRequest):
     API_KEYS_DB[new_key] = {"limit": data.limit, "used": 0, "status": "active"}
     return {"api_key": new_key, "limit": data.limit}
 
-@app.post("/v1/chat")
+@main.post("/v1/chat")
 async def chat(message: ChatRequest, x_api_key: str = Header(None)):
     if not x_api_key or x_api_key not in API_KEYS_DB:
         raise HTTPException(status_code=403, detail="Access Denied")