Update README.md

README.md

@@ -120,6 +120,7 @@ Jawaban:
 
 ### Instalasi
 
+```python
 # ==================== CELL 1: INSTALL ====================
 
 # Install library utama
@@ -134,60 +135,83 @@ print("SETELAH INSTALL SELESAI:")
 print("1. Klik Runtime → Restart Session")
 print("2. Jalankan cell berikutnya")
 print("=" * 60)
+```
 
 Penggunaan Dasar
 
 ```python
-from transformers import AutoModelForCausalLM, AutoTokenizer
+# ==================== CELL 2: TEST ORBIT-3-NANO ====================
+
+# ─── Import ───
+import os
 import torch
+import transformers
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-MODEL_NAME = "OpenPathAI/Orbit-3.2-Nano"
+# ─── Cek Versi ───
+print(f"transformers: {transformers.__version__}")
+print(f"torch: {torch.__version__}")
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+# ─── Config ───
+HF_TOKEN = "TOKEN_HF"
+MODEL_NAME = "OpenPathAI/Orbit-3-Nano"
+
+os.environ["HUGGING_FACE_HUB_TOKEN"] = HF_TOKEN
+os.environ["HF_TOKEN"] = HF_TOKEN
+
+# ==================== LOAD TOKENIZER ====================
+print("\nLoading tokenizer...")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_TOKEN)
+print("✅ Tokenizer loaded!")
+
+# ==================== LOAD MODEL ====================
+print("Loading model...")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float16,
     device_map="auto",
+    token=HF_TOKEN,
 )
-
-# Contoh dengan konteks
-context = "Virgin Australia adalah maskapai penerbangan yang berbasis di Australia. Ini memulai layanan pada 31 Agustus 2000 sebagai Virgin Blue."
-question = "Kapan Virgin Australia mulai beroperasi?"
-
-messages = [
-    {"role": "user", "content": f"Konteks:\n{context}\n\nPertanyaan:\n{question}"}
-]
-
-inputs = tokenizer.apply_chat_template(
-    messages,
-    add_generation_prompt=True,
-    return_tensors="pt",
-    return_dict=True,
-).to(model.device)
-
-outputs = model.generate(
-    **inputs,
-    max_new_tokens=512,
-    temperature=0.7,
-    top_p=0.9,
-    do_sample=True,
-    pad_token_id=tokenizer.eos_token_id,
-)
-
-response = tokenizer.decode(
-    outputs[0][inputs["input_ids"].shape[1]:],
-    skip_special_tokens=True
-)
-print(response)
-```
-
-Contoh Tanpa Konteks
-
-```python
-messages = [
-    {"role": "user", "content": "Jelaskan apa itu machine learning"}
-]
-# ... (sama seperti di atas)
+model.eval()
+print("✅ Model loaded!")
+
+# ==================== FUNGSI CHAT ====================
+def chat(pertanyaan, max_new_tokens=512, temperature=0.7):
+    """Fungsi chat untuk transformers v5."""
+    messages = [{"role": "user", "content": pertanyaan}]
+    
+    inputs = tokenizer.apply_chat_template(
+        messages,
+        add_generation_prompt=True,
+        return_tensors="pt",
+        return_dict=True,
+    ).to(model.device)
+    
+    with torch.no_grad():
+        outputs = model.generate(
+            **inputs,
+            max_new_tokens=max_new_tokens,
+            temperature=temperature,
+            top_p=0.9,
+            do_sample=True,
+            pad_token_id=tokenizer.eos_token_id,
+        )
+    
+    response = tokenizer.decode(
+        outputs[0][inputs["input_ids"].shape[1]:],
+        skip_special_tokens=True
+    )
+    return response.strip()
+
+# ==================== TEST ====================
+pertanyaan = "halo, kamu bisa bantu aku gak?"
+
+print("\n" + "=" * 60)
+print(f"📝 Pertanyaan: {pertanyaan}")
+print("=" * 60)
+print("🤖 Jawaban dari Orbit-3-Nano:")
+print(chat(pertanyaan))
+print("=" * 60)
 ```
 
 ---