Update checking2.py

checking2.py

@@ -1,103 +1,6 @@
-import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
+عالم خان|لعل خان|لعل خان|محلہ لوہارانوالہ،کالا باغ،تحصیل عیسی خیل،ضلع میانوالی|محلہ لوہارانوالہ،کالا باغ،تحصیل عیسی خیل،ضلع میانوالی|عیسی خیل,میانوالی
+v2: Roman:
+ISI KHIL,MIANWALI
 
-# ============================================================
-# CHANGE THIS TO YOUR V2 MODEL PATH
-# ============================================================
-
-MODEL_PATH = r"D:\path\to\v2_model"
-
-# ============================================================
-# LOAD MODEL
-# ============================================================
-
-print("Loading tokenizer...")
-tokenizer = AutoTokenizer.from_pretrained(
-    MODEL_PATH,
-    trust_remote_code=True
-)
-
-print("Loading model...")
-
-model = AutoModelForCausalLM.from_pretrained(
-    MODEL_PATH,
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-    device_map="auto",
-    trust_remote_code=True
-)
-
-model.eval()
-
-print("Model loaded successfully.\n")
-
-# ============================================================
-# PROMPT FORMAT
-# ============================================================
-
-def make_prompt(urdu_text):
-    return f"""### Instruction:
-Transliterate the following Urdu text into Roman Urdu.
-Output ONLY the Roman Urdu. No translation. No explanation.
-
-### Input:
-{urdu_text}
-
-### Response:
-"""
-
-# ============================================================
-# GENERATE
-# ============================================================
-
-def transliterate(urdu_text):
-
-    prompt = make_prompt(urdu_text)
-
-    inputs = tokenizer(
-        prompt,
-        return_tensors="pt"
-    ).to(model.device)
-
-    with torch.no_grad():
-        outputs = model.generate(
-            **inputs,
-            max_new_tokens=256,
-            do_sample=False,
-            temperature=0.0,
-            top_p=1.0,
-            repetition_penalty=1.05,
-            eos_token_id=tokenizer.eos_token_id,
-            pad_token_id=tokenizer.eos_token_id
-        )
-
-    result = tokenizer.decode(
-        outputs[0],
-        skip_special_tokens=True
-    )
-
-    if "### Response:" in result:
-        result = result.split("### Response:")[-1].strip()
-
-    return result
-
-
-# ============================================================
-# INTERACTIVE LOOP
-# ============================================================
-
-print("=" * 70)
-print("Tiny Aya V2 Transliteration Test")
-print("Type 'exit' to quit.")
-print("=" * 70)
-
-while True:
-
-    urdu = input("\nUrdu: ").strip()
-
-    if urdu.lower() == "exit":
-        break
-
-    roman = transliterate(urdu)
-
-    print("\nRoman:")
-    print(roman)
+v3:
+ALAM KHAN LAL KHAN LAL KHAN MOHALA LOHARAN WALA KALA BAGH TEHSIL EESSA KHAIL DISTRICT MIANWALI