Update app.py

app.py

@@ -1,16 +1,28 @@
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import os
+from huggingface_hub import login
 
 # --- Konfigurasi ---
 # Gunakan ID model yang spesifik dari Hugging Face Hub
 model_id = "google/gemma-3n-E4B-it-litert-preview"
+# Dapatkan token dari Hugging Face Space Secrets
+# Anda harus mengatur ini di pengaturan Space Anda!
+hf_token = os.getenv("HF_TOKEN")
+
+# Lakukan login menggunakan token jika tersedia
+if hf_token:
+    print("Melakukan login ke Hugging Face Hub...")
+    login(token=hf_token)
+else:
+    print("Peringatan: Secret HF_TOKEN tidak ditemukan. Proses unduh model mungkin gagal jika repo bersifat privat/gated.")
 
 # --- Muat Model dan Tokenizer ---
 # Proses ini hanya berjalan sekali saat aplikasi dimulai
 print("Memuat tokenizer...")
-# Memuat tokenizer yang sesuai dengan model
-tokenizer = AutoTokenizer.from_pretrained(model_id)
+# Memuat tokenizer yang sesuai dengan model, menggunakan token untuk akses
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
 
 print("Memuat model... Ini mungkin memakan waktu beberapa menit.")
 # Memuat model AI.
@@ -20,6 +32,7 @@ model = AutoModelForCausalLM.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
+    token=hf_token,
 )
 print("Model berhasil dimuat!")
 
@@ -84,4 +97,3 @@ chatbot_ui = gr.ChatInterface(
 # --- Menjalankan Aplikasi ---
 if __name__ == "__main__":
     chatbot_ui.launch()
-